Skip to content

Commit

Permalink
TST: parametrize in tests/plotting/test_frame.py (#37735)
Browse files Browse the repository at this point in the history
  • Loading branch information
@ivanovmg
ivanovmg committed Nov 10, 2020
1 parent 0ce9eb2 commit 365d843
Showing 1 changed file with 117 additions and 137 deletions.
254 changes: 117 additions & 137 deletions pandas/tests/plotting/test_frame.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -197,11 +197,11 @@ def test_color_single_series_list(self):
df = DataFrame({"A": [1, 2, 3]})
_check_plot_works(df.plot, color=["red"])

def test_rgb_tuple_color(self):
@pytest.mark.parametrize("color", [(1, 0, 0), (1, 0, 0, 0.5)])
def test_rgb_tuple_color(self, color):
# GH 16695
df = DataFrame({"x": [1, 2], "y": [3, 4]})
_check_plot_works(df.plot, x="x", y="y", color=(1, 0, 0))
_check_plot_works(df.plot, x="x", y="y", color=(1, 0, 0, 0.5))
_check_plot_works(df.plot, x="x", y="y", color=color)

def test_color_empty_string(self):
df = DataFrame(np.random.randn(10, 2))
Expand Down Expand Up @@ -450,11 +450,21 @@ def test_subplots(self):
for ax in axes:
assert ax.get_legend() is None

def test_groupby_boxplot_sharey(self):
@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],
Expand All @@ -463,27 +473,25 @@ def test_groupby_boxplot_sharey(self):
},
index=[0, 1, 2, 3, 4],
)

# behavior without keyword
axes = df.groupby("c").boxplot()
expected = [True, False, True, False]
self._assert_ytickslabels_visibility(axes, expected)

# set sharey=True should be identical
axes = df.groupby("c").boxplot(sharey=True)
expected = [True, False, True, False]
self._assert_ytickslabels_visibility(axes, expected)

# sharey=False, all yticklabels should be visible
axes = df.groupby("c").boxplot(sharey=False)
expected = [True, True, True, True]
axes = df.groupby("c").boxplot(**kwargs)
self._assert_ytickslabels_visibility(axes, expected)

def test_groupby_boxplot_sharex(self):
@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],
Expand All @@ -492,21 +500,7 @@ def test_groupby_boxplot_sharex(self):
},
index=[0, 1, 2, 3, 4],
)

# behavior without keyword
axes = df.groupby("c").boxplot()
expected = [True, True, True, True]
self._assert_xtickslabels_visibility(axes, expected)

# set sharex=False should be identical
axes = df.groupby("c").boxplot(sharex=False)
expected = [True, True, True, True]
self._assert_xtickslabels_visibility(axes, expected)

# sharex=True, yticklabels should be visible
# only for bottom plots
axes = df.groupby("c").boxplot(sharex=True)
expected = [False, False, True, True]
axes = df.groupby("c").boxplot(**kwargs)
self._assert_xtickslabels_visibility(axes, expected)

@pytest.mark.slow
Expand Down Expand Up @@ -565,24 +559,12 @@ def test_subplots_timeseries_y_axis(self):
}
testdata = DataFrame(data)

ax_numeric = testdata.plot(y="numeric")
assert (
ax_numeric.get_lines()[0].get_data()[1] == testdata["numeric"].values
).all()
ax_timedelta = testdata.plot(y="timedelta")
assert (
ax_timedelta.get_lines()[0].get_data()[1] == testdata["timedelta"].values
).all()
ax_datetime_no_tz = testdata.plot(y="datetime_no_tz")
assert (
ax_datetime_no_tz.get_lines()[0].get_data()[1]
== testdata["datetime_no_tz"].values
).all()
ax_datetime_all_tz = testdata.plot(y="datetime_all_tz")
assert (
ax_datetime_all_tz.get_lines()[0].get_data()[1]
== testdata["datetime_all_tz"].values
).all()
y_cols = ["numeric", "timedelta", "datetime_no_tz", "datetime_all_tz"]
for col in y_cols:
ax = testdata.plot(y=col)
result = ax.get_lines()[0].get_data()[1]
expected = testdata[col].values
assert (result == expected).all()

msg = "no numeric data to plot"
with pytest.raises(TypeError, match=msg):
Expand Down Expand Up @@ -640,7 +622,7 @@ def test_subplots_timeseries_y_axis_not_supported(self):
).all()

@pytest.mark.slow
def test_subplots_layout(self):
def test_subplots_layout_multi_column(self):
# GH 6667
df = DataFrame(np.random.rand(10, 3), index=list(string.ascii_letters[:10]))

Expand Down Expand Up @@ -673,15 +655,26 @@ def test_subplots_layout(self):
with pytest.raises(ValueError):
df.plot(subplots=True, layout=(-1, -1))

# single column
@pytest.mark.slow
@pytest.mark.parametrize(
"kwargs, expected_axes_num, expected_layout, expected_shape",
[
({}, 1, (1, 1), (1,)),
({"layout": (3, 3)}, 1, (3, 3), (3, 3)),
],
)
def test_subplots_layout_single_column(
self, kwargs, expected_axes_num, expected_layout, expected_shape
):
# GH 6667
df = DataFrame(np.random.rand(10, 1), index=list(string.ascii_letters[:10]))
axes = df.plot(subplots=True)
self._check_axes_shape(axes, axes_num=1, layout=(1, 1))
assert axes.shape == (1,)

axes = df.plot(subplots=True, layout=(3, 3))
self._check_axes_shape(axes, axes_num=1, layout=(3, 3))
assert axes.shape == (3, 3)
axes = df.plot(subplots=True, **kwargs)
self._check_axes_shape(
axes,
axes_num=expected_axes_num,
layout=expected_layout,
)
assert axes.shape == expected_shape

@pytest.mark.slow
def test_subplots_warnings(self):
Expand Down Expand Up @@ -1073,24 +1066,20 @@ def test_bar_barwidth(self):
assert r.get_height() == width

@pytest.mark.slow
def test_bar_barwidth_position(self):
@pytest.mark.parametrize(
"kwargs",
[
{"kind": "bar", "stacked": False},
{"kind": "bar", "stacked": True},
{"kind": "barh", "stacked": False},
{"kind": "barh", "stacked": True},
{"kind": "bar", "subplots": True},
{"kind": "barh", "subplots": True},
],
)
def test_bar_barwidth_position(self, kwargs):
df = DataFrame(np.random.randn(5, 5))
self._check_bar_alignment(
df, kind="bar", stacked=False, width=0.9, position=0.2
)
self._check_bar_alignment(df, kind="bar", stacked=True, width=0.9, position=0.2)
self._check_bar_alignment(
df, kind="barh", stacked=False, width=0.9, position=0.2
)
self._check_bar_alignment(
df, kind="barh", stacked=True, width=0.9, position=0.2
)
self._check_bar_alignment(
df, kind="bar", subplots=True, width=0.9, position=0.2
)
self._check_bar_alignment(
df, kind="barh", subplots=True, width=0.9, position=0.2
)
self._check_bar_alignment(df, width=0.9, position=0.2, **kwargs)

@pytest.mark.slow
def test_bar_barwidth_position_int(self):
Expand Down Expand Up @@ -1508,68 +1497,59 @@ def _check_bar_alignment(
return axes

@pytest.mark.slow
def test_bar_stacked_center(self):
@pytest.mark.parametrize(
"kwargs",
[
# stacked center
dict(kind="bar", stacked=True),
dict(kind="bar", stacked=True, width=0.9),
dict(kind="barh", stacked=True),
dict(kind="barh", stacked=True, width=0.9),
# center
dict(kind="bar", stacked=False),
dict(kind="bar", stacked=False, width=0.9),
dict(kind="barh", stacked=False),
dict(kind="barh", stacked=False, width=0.9),
# subplots center
dict(kind="bar", subplots=True),
dict(kind="bar", subplots=True, width=0.9),
dict(kind="barh", subplots=True),
dict(kind="barh", subplots=True, width=0.9),
# align edge
dict(kind="bar", stacked=True, align="edge"),
dict(kind="bar", stacked=True, width=0.9, align="edge"),
dict(kind="barh", stacked=True, align="edge"),
dict(kind="barh", stacked=True, width=0.9, align="edge"),
dict(kind="bar", stacked=False, align="edge"),
dict(kind="bar", stacked=False, width=0.9, align="edge"),
dict(kind="barh", stacked=False, align="edge"),
dict(kind="barh", stacked=False, width=0.9, align="edge"),
dict(kind="bar", subplots=True, align="edge"),
dict(kind="bar", subplots=True, width=0.9, align="edge"),
dict(kind="barh", subplots=True, align="edge"),
dict(kind="barh", subplots=True, width=0.9, align="edge"),
],
)
def test_bar_align_multiple_columns(self, kwargs):
# GH2157
df = DataFrame({"A": [3] * 5, "B": list(range(5))}, index=range(5))
self._check_bar_alignment(df, kind="bar", stacked=True)
self._check_bar_alignment(df, kind="bar", stacked=True, width=0.9)
self._check_bar_alignment(df, kind="barh", stacked=True)
self._check_bar_alignment(df, kind="barh", stacked=True, width=0.9)

@pytest.mark.slow
def test_bar_center(self):
df = DataFrame({"A": [3] * 5, "B": list(range(5))}, index=range(5))
self._check_bar_alignment(df, kind="bar", stacked=False)
self._check_bar_alignment(df, kind="bar", stacked=False, width=0.9)
self._check_bar_alignment(df, kind="barh", stacked=False)
self._check_bar_alignment(df, kind="barh", stacked=False, width=0.9)
self._check_bar_alignment(df, **kwargs)

@pytest.mark.slow
def test_bar_subplots_center(self):
df = DataFrame({"A": [3] * 5, "B": list(range(5))}, index=range(5))
self._check_bar_alignment(df, kind="bar", subplots=True)
self._check_bar_alignment(df, kind="bar", subplots=True, width=0.9)
self._check_bar_alignment(df, kind="barh", subplots=True)
self._check_bar_alignment(df, kind="barh", subplots=True, width=0.9)

@pytest.mark.slow
def test_bar_align_single_column(self):
@pytest.mark.parametrize(
"kwargs",
[
dict(kind="bar", stacked=False),
dict(kind="bar", stacked=True),
dict(kind="barh", stacked=False),
dict(kind="barh", stacked=True),
dict(kind="bar", subplots=True),
dict(kind="barh", subplots=True),
],
)
def test_bar_align_single_column(self, kwargs):
df = DataFrame(np.random.randn(5))
self._check_bar_alignment(df, kind="bar", stacked=False)
self._check_bar_alignment(df, kind="bar", stacked=True)
self._check_bar_alignment(df, kind="barh", stacked=False)
self._check_bar_alignment(df, kind="barh", stacked=True)
self._check_bar_alignment(df, kind="bar", subplots=True)
self._check_bar_alignment(df, kind="barh", subplots=True)

@pytest.mark.slow
def test_bar_edge(self):
df = DataFrame({"A": [3] * 5, "B": list(range(5))}, index=range(5))

self._check_bar_alignment(df, kind="bar", stacked=True, align="edge")
self._check_bar_alignment(df, kind="bar", stacked=True, width=0.9, align="edge")
self._check_bar_alignment(df, kind="barh", stacked=True, align="edge")
self._check_bar_alignment(
df, kind="barh", stacked=True, width=0.9, align="edge"
)

self._check_bar_alignment(df, kind="bar", stacked=False, align="edge")
self._check_bar_alignment(
df, kind="bar", stacked=False, width=0.9, align="edge"
)
self._check_bar_alignment(df, kind="barh", stacked=False, align="edge")
self._check_bar_alignment(
df, kind="barh", stacked=False, width=0.9, align="edge"
)

self._check_bar_alignment(df, kind="bar", subplots=True, align="edge")
self._check_bar_alignment(
df, kind="bar", subplots=True, width=0.9, align="edge"
)
self._check_bar_alignment(df, kind="barh", subplots=True, align="edge")
self._check_bar_alignment(
df, kind="barh", subplots=True, width=0.9, align="edge"
)
self._check_bar_alignment(df, **kwargs)

@pytest.mark.slow
def test_bar_log_no_subplots(self):
Expand Down

0 comments on commit 365d843

Please sign in to comment.