use pandas.testing.assert_frame_equal

BlueBrain · Nov 7, 2022 · 4d5cd88 · 4d5cd88
1 parent 09a5387
commit 4d5cd88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/somatic_validation/tests/test_extract_features.py b/somatic_validation/tests/test_extract_features.py
@@ -5,6 +5,7 @@
 from pathlib import Path
 
 import pandas as pd
+from pandas.testing import assert_frame_equal
 
 from ..extract_features import (
     translate_legacy_targets,
@@ -57,4 +58,4 @@ def test_extract_efeatures():
     for protocol in results:
         results_df = features_df(results, protocol)
         ground_truth_df = features_df(ground_truth, protocol)
-        assert results_df.equals(ground_truth_df)
+        assert_frame_equal(results_df, ground_truth_df)