Add test function to catch divide-by-zero warning

TDAmeritrade · Jul 25, 2024 · aea50d9 · aea50d9
1 parent f154e2e
commit aea50d9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_ostinato.py b/tests/test_ostinato.py
@@ -249,3 +249,13 @@ def test_extract_several_consensus_ostinatoed(dask_cluster):
                     Ts_ref[i][np.isfinite(Ts_ref[i])],
                     Ts_comp[i][np.isfinite(Ts_comp[i])],
                 )
+
+
+@pytest.mark.filterwarnings(
+    "error:divide by zero encountered in divide", category=RuntimeWarning
+)
+def test_ostinato_divide_by_zero():
+    Ts = [np.random.rand(n) for n in [64, 128, 256]]
+    m = 5
+    Ts[0][:m] = np.nan
+    stumpy.ostinato(Ts, m)