Fixed aamp ostinato unit test coverage

Author: seanlaw

stumpy/aamp_ostinato.py

@@ -126,7 +126,7 @@ def _get_aamp_central_motif(
         if (
             np.isclose(candidate_nns_radii.max(), bsf_radius)
             and candidate_nns_radii.mean() < bsf_nns_mean_radii
-        ):  # pragma: no cover
+        ):
             bsf_Ts_idx = Ts_idx
             bsf_subseq_idx = subseq_idx
             bsf_nns_mean_radii = candidate_nns_radii.mean()

stumpy/ostinato.py

@@ -122,7 +122,7 @@ def _get_central_motif(Ts, bsf_radius, bsf_Ts_idx, bsf_subseq_idx, m, M_Ts, Σ_T
         if (
             np.isclose(candidate_nns_radii.max(), bsf_radius)
             and candidate_nns_radii.mean() < bsf_nns_mean_radii
-        ):  # pragma: no cover
+        ):
             bsf_Ts_idx = Ts_idx
             bsf_subseq_idx = subseq_idx
             bsf_nns_mean_radii = candidate_nns_radii.mean()

tests/test_aamp_ostinato.py

@@ -29,7 +29,7 @@ def test_random_ostinato(seed):
     npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)
 
 
-@pytest.mark.parametrize("seed", [79, 109, 112, 133, 151, 161, 251, 275, 309, 355])
+@pytest.mark.parametrize("seed", [41, 88, 290, 292, 310, 328, 538, 556, 563, 570])
 def test_deterministic_ostinato(seed):
     m = 50
     np.random.seed(seed)
@@ -62,7 +62,7 @@ def test_random_ostinatoed(seed, dask_cluster):
         npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)
 
 
-@pytest.mark.parametrize("seed", [79, 109, 112, 133, 151, 161, 251, 275, 309, 355])
+@pytest.mark.parametrize("seed", [41, 88, 290, 292, 310, 328, 538, 556, 563, 570])
 def test_deterministic_ostinatoed(seed, dask_cluster):
     with Client(dask_cluster) as dask_client:
         m = 50

tests/test_gpu_aamp_ostinato.py

@@ -28,7 +28,7 @@ def test_random_gpu_aamp_ostinato(seed):
     npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)
 
 
-@pytest.mark.parametrize("seed", [79, 109])
+@pytest.mark.parametrize("seed", [41, 88])
 def test_deterministic_gpu_aamp_ostinato(seed):
     m = 50
     np.random.seed(seed)