1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
| """ Test cases for DataFrame.plot """
|
| import pytest
|
| import pandas.util._test_decorators as td
|
| from pandas import DataFrame
| from pandas.tests.plotting.common import TestPlotBase
|
|
| @td.skip_if_no_mpl
| class TestDataFramePlotsGroupby(TestPlotBase):
| def _assert_ytickslabels_visibility(self, axes, expected):
| for ax, exp in zip(axes, expected):
| self._check_visible(ax.get_yticklabels(), visible=exp)
|
| def _assert_xtickslabels_visibility(self, axes, expected):
| for ax, exp in zip(axes, expected):
| self._check_visible(ax.get_xticklabels(), visible=exp)
|
| @pytest.mark.parametrize(
| "kwargs, expected",
| [
| # behavior without keyword
| ({}, [True, False, True, False]),
| # set sharey=True should be identical
| ({"sharey": True}, [True, False, True, False]),
| # sharey=False, all yticklabels should be visible
| ({"sharey": False}, [True, True, True, True]),
| ],
| )
| def test_groupby_boxplot_sharey(self, kwargs, expected):
| # https://github.com/pandas-dev/pandas/issues/20968
| # sharey can now be switched check whether the right
| # pair of axes is turned on or off
| df = DataFrame(
| {
| "a": [-1.43, -0.15, -3.70, -1.43, -0.14],
| "b": [0.56, 0.84, 0.29, 0.56, 0.85],
| "c": [0, 1, 2, 3, 1],
| },
| index=[0, 1, 2, 3, 4],
| )
| axes = df.groupby("c").boxplot(**kwargs)
| self._assert_ytickslabels_visibility(axes, expected)
|
| @pytest.mark.parametrize(
| "kwargs, expected",
| [
| # behavior without keyword
| ({}, [True, True, True, True]),
| # set sharex=False should be identical
| ({"sharex": False}, [True, True, True, True]),
| # sharex=True, xticklabels should be visible
| # only for bottom plots
| ({"sharex": True}, [False, False, True, True]),
| ],
| )
| def test_groupby_boxplot_sharex(self, kwargs, expected):
| # https://github.com/pandas-dev/pandas/issues/20968
| # sharex can now be switched check whether the right
| # pair of axes is turned on or off
|
| df = DataFrame(
| {
| "a": [-1.43, -0.15, -3.70, -1.43, -0.14],
| "b": [0.56, 0.84, 0.29, 0.56, 0.85],
| "c": [0, 1, 2, 3, 1],
| },
| index=[0, 1, 2, 3, 4],
| )
| axes = df.groupby("c").boxplot(**kwargs)
| self._assert_xtickslabels_visibility(axes, expected)
|
|
