fix settings for row tests

lenskit · May 27, 2024 · 7bf0046 · 7bf0046
1 parent 6078a3d
commit 7bf0046
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_matrix_rows.py b/tests/test_matrix_rows.py
@@ -39,6 +39,7 @@ def test_sparse_stats(tensor):
     assert tots.numpy()[mask] == approx(sums.numpy()[mask])
 
 
+@settings(deadline=500, suppress_health_check=[HealthCheck.too_slow])
 @given(sparse_tensors())
 def test_sparse_mean_center(tensor):
     nr, nc = tensor.shape
@@ -54,7 +55,7 @@ def test_sparse_mean_center(tensor):
         assert nr == approx(tr - means[i].numpy())
 
 
-@settings(deadline=500)
+@settings(deadline=500, suppress_health_check=[HealthCheck.too_slow])
 @given(sparse_tensors())
 def test_sparse_unit_norm(tensor):
     nr, nc = tensor.shape