fix tests

Author: R-Palazzo

sdmetrics/column_pairs/statistical/correlation_similarity.py

@@ -65,6 +65,12 @@ def compute_breakdown(
                 The values from the real dataset.
             synthetic_data (Union[numpy.ndarray, pandas.Series]):
                 The values from the synthetic dataset.
+            coefficient (str):
+                The correlation coefficient to use. Either 'Pearson' or 'Spearman'.
+                Default is 'Pearson'.
+            real_correlation_threshold (float):
+                The minimum absolute correlation value for the real data to be considered
+                correlated. Default is 0.
 
         Returns:
             dict:

tests/unit/column_pairs/statistical/test_correlation_similarity.py

@@ -129,7 +129,7 @@ def test_compute_breakdown_constant_input(self):
     def test_compute_breakdown_with_real_correlation_threshold(
         self, real_correlation_threshold, score
     ):
-        """Test the ``compute_breakdown`` method with real correlation threshold.
+        """Test the ``compute_breakdown`` method with `real_correlation_threshold`.
 
         In this test, real data has a correlation of 0.498212 and synthetic data
         has a correlation of 0.3.
@@ -154,7 +154,11 @@ def test_compute_breakdown_with_real_correlation_threshold(
         )
 
         # Assert
-        assert result['score'] == score if not np.isnan(score) else np.isnan(result['score'])
+        assert (
+            np.isclose(result['score'], score, atol=1e-6)
+            if not np.isnan(score)
+            else np.isnan(result['score'])
+        )
 
     def test_compute_breakdown_invalid_real_correlation_threshold(self):
         """Test an error is thrown when an invalid `real_correlation_threshold` is passed."""