From d9f075498eb1589610a4b394c4f8d843831ea01f Mon Sep 17 00:00:00 2001 From: Xinlan Emily Hu Date: Tue, 9 Apr 2024 19:34:36 -0400 Subject: [PATCH] round outputs in tests --- feature_engine/testing/test_feature_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/feature_engine/testing/test_feature_metrics.py b/feature_engine/testing/test_feature_metrics.py index 40258a2d..2a72146d 100644 --- 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