use the testing helper from our own module now

pylhc · Nov 14, 2024 · aa85414 · aa85414
1 parent 3d27744
commit aa85414
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 7 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -18,8 +18,7 @@
     NonStringColumnNameError,
     SpaceinColumnNameError,
 )
-
-from .conftest import assert_tfs_frame_equal
+from tfs.testing import assert_tfs_frame_equal
 
 
 class TestWrites: