Fix tests

adap · Aug 11, 2024 · aa69c2e · aa69c2e
1 parent dc8b83c
commit aa69c2e
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 5 deletions.
diff --git a/src/py/flwr/server/strategy/bulyan_test.py b/src/py/flwr/server/strategy/bulyan_test.py
@@ -129,4 +129,5 @@ def test_aggregate_fit() -> None:
         actual_list = parameters_to_ndarrays(actual_aggregated)
         actual = actual_list[0]
         assert (actual == expected[0]).all()
-    raise ValueError("aggregate_fit function returned None")
+    else:
+        raise ValueError("aggregate_fit function returned None")
diff --git a/src/py/flwr/server/strategy/fedadagrad_test.py b/src/py/flwr/server/strategy/fedadagrad_test.py
@@ -84,4 +84,5 @@ def test_aggregate_fit() -> None:
         actual_list = parameters_to_ndarrays(actual_aggregated)
         actual = actual_list[0]
         assert (actual == expected[0]).all()
-    raise ValueError("aggregate_fit function returned None")
+    else:
+        raise ValueError("aggregate_fit function returned None")
diff --git a/src/py/flwr/server/strategy/fedmedian_test.py b/src/py/flwr/server/strategy/fedmedian_test.py
@@ -198,4 +198,5 @@ def test_aggregate_fit() -> None:
         actual_list = parameters_to_ndarrays(actual_aggregated)
         actual = actual_list[0]
         assert (actual == expected[0]).all()
-    raise ValueError("aggregate_fit function returned None")
+    else:
+        raise ValueError("aggregate_fit function returned None")
diff --git a/src/py/flwr/server/strategy/krum_test.py b/src/py/flwr/server/strategy/krum_test.py
@@ -199,4 +199,5 @@ def test_aggregate_fit() -> None:
         actual_list = parameters_to_ndarrays(actual_aggregated)
         actual = actual_list[0]
         assert (actual == expected[0]).all()
-    raise ValueError("aggregate_fit function returned None")
+    else:
+        raise ValueError("aggregate_fit function returned None")
diff --git a/src/py/flwr/server/strategy/multikrum_test.py b/src/py/flwr/server/strategy/multikrum_test.py
@@ -98,4 +98,5 @@ def test_aggregate_fit() -> None:
         actual_list = parameters_to_ndarrays(actual_aggregated)
         actual = actual_list[0]
         assert (actual == expected[0]).all()
-    raise ValueError("aggregate_fit function returned None")
+    else:
+        raise ValueError("aggregate_fit function returned None")