Skip to content

Commit 17dc5b9

Browse files
author
tp
committed
changed according to comments
1 parent 14d388e commit 17dc5b9

File tree

2 files changed

+17
-7
lines changed

2 files changed

+17
-7
lines changed

pandas/conftest.py

+17
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,23 @@ def spmatrix(request):
6060
return getattr(sparse, request.param + '_matrix')
6161

6262

63+
@pytest.fixture(params=[0, 1],
64+
ids=lambda x: "axis {}".format(x))
65+
def axis(request):
66+
"""
67+
Fixture for returning the axis numbers of a dataframe.
68+
"""
69+
return request.param
70+
71+
72+
@pytest.fixture(params=[0], ids=lambda x: "axis {}".format(x))
73+
def axis_series(request):
74+
"""
75+
Fixture for returning the axis numbers of a series.
76+
"""
77+
return request.param
78+
79+
6380
@pytest.fixture
6481
def ip():
6582
"""

pandas/tests/generic/test_label_or_level_utils.py

-7
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,6 @@ def assert_level_reference(frame, levels, axis):
7676

7777
# DataFrame
7878
# ---------
79-
@pytest.mark.parametrize('axis', [0, 1])
8079
def test_is_level_or_label_reference_df_simple(df_levels, axis):
8180

8281
# Compute expected labels and levels
@@ -91,7 +90,6 @@ def test_is_level_or_label_reference_df_simple(df_levels, axis):
9190
assert_label_reference(df_levels, expected_labels, axis=axis)
9291

9392

94-
@pytest.mark.parametrize('axis', [0, 1])
9593
def test_is_level_reference_df_ambig(df_ambig, axis):
9694

9795
# Transpose frame if axis == 1
@@ -165,7 +163,6 @@ def test_is_label_or_level_reference_panel_error(panel):
165163

166164
# DataFrame
167165
# ---------
168-
@pytest.mark.parametrize('axis', [0, 1])
169166
def test_check_label_or_level_ambiguity_df(df_ambig, axis):
170167

171168
# Transpose frame if axis == 1
@@ -264,7 +261,6 @@ def assert_level_values(frame, levels, axis):
264261

265262
# DataFrame
266263
# ---------
267-
@pytest.mark.parametrize('axis', [0, 1])
268264
def test_get_label_or_level_values_df_simple(df_levels, axis):
269265

270266
# Compute expected labels and levels
@@ -279,7 +275,6 @@ def test_get_label_or_level_values_df_simple(df_levels, axis):
279275
assert_level_values(df_levels, expected_levels, axis=axis)
280276

281277

282-
@pytest.mark.parametrize('axis', [0, 1])
283278
def test_get_label_or_level_values_df_ambig(df_ambig, axis):
284279

285280
# Transpose frame if axis == 1
@@ -300,7 +295,6 @@ def test_get_label_or_level_values_df_ambig(df_ambig, axis):
300295
assert_label_values(df_ambig, ['L3'], axis=axis)
301296

302297

303-
@pytest.mark.parametrize('axis', [0, 1])
304298
def test_get_label_or_level_values_df_duplabels(df_duplabels, axis):
305299

306300
# Transpose frame if axis == 1
@@ -383,7 +377,6 @@ def assert_levels_dropped(frame, levels, axis):
383377

384378
# DataFrame
385379
# ---------
386-
@pytest.mark.parametrize('axis', [0, 1])
387380
def test_drop_labels_or_levels_df(df_levels, axis):
388381

389382
# Compute expected labels and levels

0 commit comments

Comments
 (0)