Skip to content

Commit ef30b96

Browse files
committed
Added ref cache length check
1 parent ffa468c commit ef30b96

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

tests/test_cache.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import numba
12
import numpy as np
23

34
from stumpy import cache, stump
@@ -9,23 +10,25 @@ def test_cache_get_njit_funcs():
910

1011

1112
def test_cache_save_after_clear():
12-
cache.save()
13-
1413
T = np.random.rand(10)
1514
m = 3
16-
stump(T, m)
1715

16+
cache.save()
17+
stump(T, m)
1818
ref_cache = cache._get_cache()
1919

20+
if numba.config.DISABLE_JIT:
21+
assert len(ref_cache) == 0
22+
else:
23+
assert len(ref_cache) > 0
24+
2025
cache.clear()
21-
# testing cache._clear()
2226
assert len(cache._get_cache()) == 0
2327

2428
cache.save()
2529
stump(T, m)
2630
comp_cache = cache._get_cache()
2731

28-
# testing cache._save() after cache._clear()
2932
assert sorted(ref_cache) == sorted(comp_cache)
3033

3134
cache.clear()

0 commit comments

Comments
 (0)