Skip to content

Commit 904229a

Browse files
committed
squash: tests: result/expected, Base, fixture scope
1 parent 5d54674 commit 904229a

File tree

1 file changed

+18
-19
lines changed

1 file changed

+18
-19
lines changed

pandas/tests/io/test_pytables.py

Lines changed: 18 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -5170,7 +5170,7 @@ def test_dst_transitions(self):
51705170
assert_frame_equal(result, df)
51715171

51725172

5173-
@pytest.fixture(scope='module')
5173+
@pytest.fixture()
51745174
def pytables_hdf5_file():
51755175
"""Use PyTables to create a simple HDF5 file."""
51765176

@@ -5203,36 +5203,35 @@ def pytables_hdf5_file():
52035203
return tmpfilepath, objectname, pd.DataFrame(testsamples)
52045204

52055205

5206-
class TestReadPyTablesHDF5(Base):
5206+
class TestReadPyTablesHDF5:
52075207
"""
52085208
A group of tests which covers reading HDF5 files written by plain PyTables
52095209
(not written by pandas).
52105210
"""
52115211

52125212
def test_read_complete(self, pytables_hdf5_file):
5213-
path, objname, expected_df = pytables_hdf5_file
5214-
assert_frame_equal(pd.read_hdf(path, key=objname), expected_df)
5213+
path, objname, df = pytables_hdf5_file
5214+
result = pd.read_hdf(path, key=objname)
5215+
expected = df
5216+
assert_frame_equal(result, expected)
52155217

52165218
def test_read_with_start(self, pytables_hdf5_file):
5217-
path, objname, expected_df = pytables_hdf5_file
5219+
path, objname, df = pytables_hdf5_file
52185220
# This is a regression test for pandas-dev/pandas/issues/11188
5219-
assert_frame_equal(
5220-
pd.read_hdf(path, key=objname, start=1),
5221-
expected_df[1:].reset_index(drop=True)
5222-
)
5221+
result = pd.read_hdf(path, key=objname, start=1)
5222+
expected = df[1:].reset_index(drop=True)
5223+
assert_frame_equal(result, expected)
52235224

52245225
def test_read_with_stop(self, pytables_hdf5_file):
5225-
path, objname, expected_df = pytables_hdf5_file
5226+
path, objname, df = pytables_hdf5_file
52265227
# This is a regression test for pandas-dev/pandas/issues/11188
5227-
assert_frame_equal(
5228-
pd.read_hdf(path, key=objname, stop=1),
5229-
expected_df[:1].reset_index(drop=True)
5230-
)
5228+
result = pd.read_hdf(path, key=objname, stop=1)
5229+
expected = df[:1].reset_index(drop=True)
5230+
assert_frame_equal(result, expected)
52315231

52325232
def test_read_with_startstop(self, pytables_hdf5_file):
5233-
path, objname, expected_df = pytables_hdf5_file
5233+
path, objname, df = pytables_hdf5_file
52345234
# This is a regression test for pandas-dev/pandas/issues/11188
5235-
assert_frame_equal(
5236-
pd.read_hdf(path, key=objname, start=1, stop=2),
5237-
expected_df[1:2].reset_index(drop=True)
5238-
)
5235+
result = pd.read_hdf(path, key=objname, start=1, stop=2)
5236+
expected = df[1:2].reset_index(drop=True)
5237+
assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)