small change in testing the estimate_b

swiss-seismological-service · Jun 5, 2024 · f9d16a2 · f9d16a2
1 parent 710d308
commit f9d16a2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/seismostats/analysis/tests/test_estimate_beta.py b/seismostats/analysis/tests/test_estimate_beta.py
@@ -49,9 +49,7 @@ def test_estimate_b(
     )
 
     # assert that the correct number of values are returned
-    assert np.size(out) == 1 + return_std + (
-        return_n * (1 - (method == "utsu")) * (1 - (method == "tinti"))
-    )
+    assert np.size(out) == 1 + return_std + (return_n * (method == "positive"))
 
     # test that uncorrect binninng leads to error
     with pytest.raises(AssertionError) as excinfo: