From de17b609a739ca861b56af1d95f141d831826be7 Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Tue, 21 May 2024 12:38:30 +0200 Subject: [PATCH] Fix tests --- datasets/flwr_datasets/telemetry_test.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/datasets/flwr_datasets/telemetry_test.py b/datasets/flwr_datasets/telemetry_test.py index 2f82d0ec127c..676b857fe0ae 100644 --- a/datasets/flwr_datasets/telemetry_test.py +++ b/datasets/flwr_datasets/telemetry_test.py @@ -26,7 +26,7 @@ class TelemetryTest(unittest.TestCase): """Tests for the telemetry module.""" - @mock.patch("flwr_datasets.telemetry.FLWR_DATASETS_TELEMETRY_ENABLED", "1") + @mock.patch("flwr_datasets.telemetry.FLWR_TELEMETRY_ENABLED", "1") def test_event(self) -> None: """Test if sending works against the actual API.""" # Prepare @@ -39,7 +39,7 @@ def test_event(self) -> None: # Assert self.assertEqual(actual, expected) - @mock.patch("flwr_datasets.telemetry.FLWR_DATASETS_TELEMETRY_ENABLED", "1") + @mock.patch("flwr_datasets.telemetry.FLWR_TELEMETRY_ENABLED", "1") def test_not_blocking(self) -> None: """Test if the code is blocking. @@ -58,7 +58,7 @@ def test_not_blocking(self) -> None: # Assert self.assertLess(duration_actual, duration_max) - @mock.patch("flwr.common.telemetry.FLWR_TELEMETRY_ENABLED", "0") + @mock.patch("flwr_datasets.telemetry.FLWR_TELEMETRY_ENABLED", "0") def test_telemetry_disabled(self) -> None: """Test opt-out.""" # Prepare @@ -102,10 +102,14 @@ def _new_failing_get_home() -> None: # Execute with mock.patch( - "flwr.common.telemetry._get_home", + "flwr_datasets.telemetry._get_home", new_callable=new_callable, ): source_id = _get_source_id() # Assert self.assertEqual(source_id, except_value) + + +if __name__ == "__main__": + unittest.main()