diff --git a/tests/test_collection.py b/tests/test_collection.py index 42eca046..938419b9 100644 --- a/tests/test_collection.py +++ b/tests/test_collection.py @@ -5,8 +5,9 @@ from tfs import read_tfs from tfs.collection import Tfs, TfsCollection from tfs.frame import TfsDataFrame +from tfs.testing import assert_tfs_frame_equal -from .conftest import INPUTS_DIR, assert_tfs_frame_equal +from .conftest import INPUTS_DIR class CollectionTest(TfsCollection): diff --git a/tests/test_compression.py b/tests/test_compression.py index b2637327..d10f5370 100644 --- a/tests/test_compression.py +++ b/tests/test_compression.py @@ -7,9 +7,10 @@ import pytest from tfs.reader import read_headers, read_tfs +from tfs.testing import assert_tfs_frame_equal from tfs.writer import write_tfs -from .conftest import INPUTS_DIR, assert_tfs_frame_equal +from .conftest import INPUTS_DIR SUPPORTED_EXTENSIONS: tuple[str] = ["gz", "bz2", "zip", "xz", "zst", "tar", "tar.gz"] # through pandas diff --git a/tests/test_hdf.py b/tests/test_hdf.py index 42031e14..8ebbcfcc 100644 --- a/tests/test_hdf.py +++ b/tests/test_hdf.py @@ -4,8 +4,7 @@ import pytest from tfs import TfsDataFrame, read_hdf, read_tfs, write_hdf - -from .conftest import assert_tfs_frame_equal +from tfs.testing import assert_tfs_frame_equal class TestHDF: diff --git a/tests/test_reader.py b/tests/test_reader.py index 62e75454..ba121542 100644 --- a/tests/test_reader.py +++ b/tests/test_reader.py @@ -9,9 +9,10 @@ from tfs.constants import HEADER from tfs.errors import AbsentColumnNameError, AbsentColumnTypeError, UnknownTypeIdentifierError from tfs.reader import read_headers, read_tfs +from tfs.testing import assert_tfs_frame_equal from tfs.writer import write_tfs -from .conftest import INPUTS_DIR, assert_tfs_frame_equal +from .conftest import INPUTS_DIR class TestRead: diff --git a/tests/test_writer.py b/tests/test_writer.py index 6a54e26c..46a13283 100644 --- a/tests/test_writer.py +++ b/tests/test_writer.py @@ -18,8 +18,7 @@ NonStringColumnNameError, SpaceinColumnNameError, ) - -from .conftest import assert_tfs_frame_equal +from tfs.testing import assert_tfs_frame_equal class TestWrites: