fixing tests

Author: shakedzy

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## 0.7.10
 * Fix a bug in `model_utils.metric_graph`
+* Fix failing tests
 
 ## 0.7.9
 * Fixing `nominal.associations(plot=False)` not working as expected on Jupyter-based notebooks (issues [#167](https://github.com/shakedzy/dython/issues/167) & [#168](https://github.com/shakedzy/dython/issues/168))

tests/test_nominal/test_associations.py

@@ -122,9 +122,8 @@ def test_datetime_data():
         data=[[1.0, 1.0, -1.0], [1.0, 1.0, -1.0], [-1.0, -1.0, 1.0]],
     )
     corr = associations(df, plot=False)["corr"]
-    assert corr.compare(
-        correct_corr
-    ).empty, f"datetime associations are incorrect. Test should have returned an empty dataframe, received: {corr.head()}"
+    assert np.allclose(corr, correct_corr, atol=1e-10, rtol=1e-10), \
+          f"datetime associations are incorrect. Test should have returned an empty dataframe, received: {corr.head()}"
 
 
 def test_category_nan_replace(iris_df):

tests/test_nominal/test_associations_parallel.py

@@ -154,6 +154,5 @@ def test_datetime_data():
     corr = associations(
         df, plot=False, multiprocessing=True, max_cpu_cores=MAX_CORE_COUNT
     )["corr"]
-    assert corr.compare(
-        correct_corr
-    ).empty, f"datetime associations are incorrect. Test should have returned an empty dataframe, received: {corr.head()}"
+    assert np.allclose(corr, correct_corr, atol=1e-10, rtol=1e-10), \
+          f"datetime associations are incorrect. Test should have returned an empty dataframe, received: {corr.head()}"