test: round to 2 decimal places

Author: sahiljhawar

tests/io/kp/test_kp_niemegk.py

@@ -143,12 +143,12 @@ def test_process_single_file(self):
             assert len(df) == 2
 
             expected_values = [
-                3.667,
-                2.333,
+                3.67,
+                2.33,
             ]
 
             for actual, expected in zip(df["kp"].values, expected_values):
-                assert np.isclose(actual, expected, atol=0.001)
+                assert np.isclose(actual, expected, atol=0.0)
 
         finally:
             shutil.rmtree(temp_dir, ignore_errors=True)

tests/io/kp/test_kp_swpc.py

@@ -92,9 +92,11 @@ def test_read_with_download(self, kp_swpc_instance):
         assert data["kp"].max() <= 9
         assert data.index.tzinfo == timezone.utc
 
-        decimals = (data["kp"] % 1).unique()
+        decimals = [round(x, 2) for x in (data["kp"] % 1).unique()]
         for decimal in decimals:
-            assert np.isclose(decimal, 0) or np.isclose(decimal, 1 / 3) or np.isclose(decimal, 2 / 3)
+            assert (
+                np.isclose(decimal, 0) or np.isclose(decimal, round(1 / 3, 2)) or np.isclose(decimal, round(2 / 3, 2))
+            )
 
     def test_read_exceeding_three_days(self, kp_swpc_instance):
         start_time = datetime.now(timezone.utc)