From a7f6da00f64bae850eb739e0f5c33733661e84f5 Mon Sep 17 00:00:00 2001 From: Nicolas Date: Mon, 25 Mar 2024 18:14:42 +0100 Subject: [PATCH] introducing allclose assertion to make it pass --- tests/test_model_regression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_model_regression.py b/tests/test_model_regression.py index 81ec8d0..0e1d556 100644 --- a/tests/test_model_regression.py +++ b/tests/test_model_regression.py @@ -98,7 +98,7 @@ def test_video_prediction_tracking(expected_results, test_input, video_number): output_file_path = OUTPUT_FOLDER / f"{video_number}_test_results.txt" np.savetxt(output_file_path, combined_array, delimiter=" ", fmt="%s") - np.array_equal(expected_results, combined_array) + assert np.allclose(expected_results, combined_array) # Remove the file if the test is successful output_file_path.unlink()