Initial commit - add failing tests cases

Author: seanlaw

tests/test_precision.py

@@ -2,13 +2,12 @@
 from unittest.mock import patch
 
 import naive
-import numba
 import numpy as np
 import numpy.testing as npt
 import pytest
 from numba import cuda
 
-from stumpy import cache, config, core, fastmath, rng, sdp
+from stumpy import config, core, rng, sdp
 
 if cuda.is_available():
     from stumpy.gpu_stump import gpu_stump
@@ -118,18 +117,19 @@ def test_calculate_squared_distance():
                 npt.assert_almost_equal(ref, comp, decimal=14)
 
 
-def test_snippets():
+@pytest.mark.parametrize(
+    "seed, m, k, s", [(2135137202, 10, 3, 3), (2636, 9, 3, 3), (332, 10, 3, 3)]
+)
+def test_snippets(seed, m, k, s):
     # This test function raises an error if there is a considerable loss of precision
     # that violates the symmetry property of a distance measure.
-    m = 10
-    k = 3
-    s = 3
-    with rng.fix_seed(332):
-        T = rng.RNG.uniform(-1000.0, 1000.0, [64])
+    with rng.fix_seed(seed):
+        T = rng.RNG.uniform(-1000, 1000, [64]).astype(np.float64)
 
         isconstant_custom_func = functools.partial(
             naive.isconstant_func_stddev_threshold, quantile_threshold=0.05
         )
+
         (
             ref_snippets,
             ref_indices,
@@ -138,9 +138,12 @@ def test_snippets():
             ref_areas,
             ref_regimes,
         ) = naive.mpdist_snippets(
-            T, m, k, s=s, mpdist_T_subseq_isconstant=isconstant_custom_func
+            T,
+            m,
+            k,
+            s=s,
+            mpdist_T_subseq_isconstant=isconstant_custom_func,
         )
-
         (
             cmp_snippets,
             cmp_indices,
@@ -150,29 +153,6 @@ def test_snippets():
             cmp_regimes,
         ) = snippets(T, m, k, s=s, mpdist_T_subseq_isconstant=isconstant_custom_func)
 
-        if (
-            not np.allclose(ref_snippets, cmp_snippets) and not numba.config.DISABLE_JIT
-        ):  # pragma: no cover
-            # Revise fastmath flags by removing reassoc (to improve precision),
-            # recompile njit functions, and re-compute snippets.
-            fastmath._set(
-                "core",
-                "_calculate_squared_distance",
-                {"nsz", "arcp", "contract", "afn"},
-            )
-            cache._recompile()
-
-            (
-                cmp_snippets,
-                cmp_indices,
-                cmp_profiles,
-                cmp_fractions,
-                cmp_areas,
-                cmp_regimes,
-            ) = snippets(
-                T, m, k, s=s, mpdist_T_subseq_isconstant=isconstant_custom_func
-            )
-
         npt.assert_almost_equal(
             ref_snippets, cmp_snippets, decimal=config.STUMPY_TEST_PRECISION
         )
@@ -190,11 +170,6 @@ def test_snippets():
         )
         npt.assert_almost_equal(ref_regimes, cmp_regimes)
 
-    if not numba.config.DISABLE_JIT:  # pragma: no cover
-        # Revert fastmath flag back to their default values
-        fastmath._reset("core", "_calculate_squared_distance")
-        cache._recompile()
-
 
 @pytest.mark.filterwarnings("ignore", category=NumbaPerformanceWarning)
 @patch("stumpy.config.STUMPY_THREADS_PER_BLOCK", TEST_THREADS_PER_BLOCK)