improve

ITISFoundation · Dec 10, 2024 · d6ad889 · d6ad889
1 parent 8210136
commit d6ad889
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/pytest-simcore/src/pytest_simcore/helpers/autoscaling.py b/packages/pytest-simcore/src/pytest_simcore/helpers/autoscaling.py
@@ -18,7 +18,7 @@
 
 def assert_cluster_state(
     spied_cluster_analysis: MockType, *, expected_calls: int, expected_num_machines: int
-) -> None:
+) -> Cluster:
     assert spied_cluster_analysis.call_count > 0
 
     assert isinstance(spied_cluster_analysis.spy_return, Cluster)
@@ -27,6 +27,7 @@ def assert_cluster_state(
         == expected_num_machines
     )
     print("current cluster state:", spied_cluster_analysis.spy_return)
+    return spied_cluster_analysis.spy_return
 
 
 def create_fake_association(