fixup

Author: dweindl

src/ccompass/core.py

@@ -826,7 +826,6 @@ def ndarray_constructor(loader, node):
             def series_constructor(loader, node):
                 """Custom YAML constructor for pandas Series."""
                 file_path = temp_dir / loader.construct_scalar(node)
-                print(file_path, type(file_path))
                 df = pd.read_csv(
                     file_path,
                     sep="\t",
@@ -835,7 +834,6 @@ def series_constructor(loader, node):
                     float_precision="round_trip",
                 )
                 assert df.shape[1] == 1
-                print(df.iloc[:, 0], type(df.iloc[:, 0]))
                 return df.iloc[:, 0]
 
             def tuple_constructor(loader, node):

tests/test_session_model.py

@@ -22,7 +22,7 @@ def test_serialization():
     assert_session_equal(session, session2)
 
 
-def assert_equal(obj1, obj2):
+def assert_equal(obj1, obj2) -> None:
     """Check if two objects are equal."""
     if isinstance(obj1, pydantic.BaseModel):
         assert isinstance(obj2, pydantic.BaseModel)
@@ -37,9 +37,16 @@ def assert_equal(obj1, obj2):
         for i in range(len(obj1)):
             assert_equal(obj1[i], obj2[i])
     elif isinstance(obj1, pd.DataFrame):
+        assert isinstance(obj2, pd.DataFrame)
+        if obj1.empty and obj2.empty:
+            # if both are empty, we don't compare columns
+            return
         pd.testing.assert_frame_equal(obj1, obj2, check_dtype=False)
     elif isinstance(obj1, pd.Series):
-        pd.testing.assert_series_equal(obj1, obj2)
+        assert isinstance(obj2, pd.Series), f"{obj1} != {obj2}"
+        pd.testing.assert_series_equal(
+            obj1, obj2, atol=1e-14, rtol=1e-14, check_dtype=False
+        )
     elif isinstance(obj1, np.ndarray):
         np.testing.assert_almost_equal(obj1, obj2)
     elif isinstance(obj1, float) and pd.isna(obj1):