round outputs in tests

Watts-Lab · Apr 9, 2024 · d9f0754 · d9f0754
1 parent 5d1ccaa
commit d9f0754
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/feature_engine/testing/test_feature_metrics.py b/feature_engine/testing/test_feature_metrics.py
@@ -48,7 +48,7 @@ def test_conv_unit_equality(conversation_num, conversation_rows):
         expected = row['expected_value']
 
     try:
-        assert actual == expected
+        assert round(actual, 3) == round(expected, 3)
     except AssertionError:
         expected_out = expected
         actual_out = actual