From eabb7582eebb43f9c32431a02929e4a4e3eb53c3 Mon Sep 17 00:00:00 2001 From: Marta Leszczynska Date: Tue, 5 Sep 2023 16:20:33 +0200 Subject: [PATCH] fix --- unit_tests/test_02_general.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unit_tests/test_02_general.py b/unit_tests/test_02_general.py index 34e32bd..a2a0929 100644 --- a/unit_tests/test_02_general.py +++ b/unit_tests/test_02_general.py @@ -13,8 +13,8 @@ def test_on_example(): assert summary_stats.loc["count", column_name] == len(dataset) assert summary_stats.loc["min", column_name] == 1 assert summary_stats.loc["max", column_name] == 5 - assert round(summary_stats.loc["mean", column_name]) == round(sum(dataset) \ - / len(dataset)) + assert round(summary_stats.loc["mean", column_name]) \ + == round(sum(dataset) / len(dataset)) assert round(summary_stats.loc["std", column_name], 2) == 1.37 assert round(summary_stats.loc["25%", column_name], 2) == 1.75 assert round(summary_stats.loc["50%", column_name], 2) == 3.00