Fix pandas formatting for a few minor versions only

mwtoews · Sep 27, 2023 · 5df78c5 · 5df78c5
1 parent 48dc205
commit 5df78c5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -25,7 +25,7 @@
 from swn.compat import ignore_shapely_warnings_for_object_array
 
 
-PANDAS_MAJOR_VERSION = int(pd.__version__[0:pd.__version__.index(".")])
+PANDAS_VESRSION_TUPLE = tuple(int(x) for x in re.findall(r"\d+", pd.__version__))
 
 datadir = Path("tests") / "data"
 

diff --git a/tests/test_file.py b/tests/test_file.py
@@ -9,7 +9,7 @@
 
 import swn
 
-from .conftest import PANDAS_MAJOR_VERSION, datadir
+from .conftest import PANDAS_VESRSION_TUPLE, datadir
 
 
 # same valid network used in test_basic
@@ -150,7 +150,7 @@ def test_read_write_formatted_frame(tmp_path):
     lines = fname.read_text().splitlines()
     assert len(lines) == 7
     # check first line, space between object columns differ between versions!
-    if PANDAS_MAJOR_VERSION >= 2:
+    if (2, 0, 0) <= PANDAS_VESRSION_TUPLE <= (2, 0, 2):
         expected = "#      value1  value2  value3"
     else:
         expected = "#      value1  value2 value3"