Update test_experimenter.py

ImperialCollegeLondon · Mar 16, 2024 · 4f070f7 · 4f070f7
1 parent 48e8863
commit 4f070f7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_experimenter.py b/tests/test_experimenter.py
@@ -5,6 +5,7 @@
 from swmmanywhere import parameters
 from swmmanywhere.paper import experimenter
 
+
 def assert_close(a: float, b: float, rtol: float = 1e-3) -> None:
     """Assert that two floats are close."""
     assert np.isclose(a, b, rtol=rtol).all()
@@ -27,7 +28,9 @@ def test_generate_samples():
                             seed = 1,
                             groups = False)
     assert len(samples) == 48
-    assert set([x['param'] for x in samples]) == {'min_v','max_v','chahinian_slope_scaling'}
+    assert set([x['param'] for x in samples]) == {'min_v',
+                                                  'max_v',
+                                                  'chahinian_slope_scaling'}
     assert_close(samples[0]['value'], 0.31093)
 
     samples = experimenter.generate_samples(N = 2,