Skip to content

Commit bee3b63

Browse files
committed
add test function to check if cache can be saved after cache._clear()
1 parent 0097953 commit bee3b63

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

tests/test_cache.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
pytest.skip("Skipping Tests JIT is disabled", allow_module_level=True)
99

1010

11-
def test_cache():
11+
def test_cache_save():
1212
def get_cache_fnames_ref():
1313
cache._clear()
1414
cache._enable()
@@ -40,3 +40,18 @@ def get_cache_fnames_comp():
4040

4141
assert sorted(ref_data) == sorted(comp_data)
4242
assert set(ref_index).issubset(comp_index)
43+
44+
45+
def test_cache_save_after_clear():
46+
T = np.random.rand(10)
47+
m = 3
48+
stump(T, m)
49+
50+
cache._save()
51+
ref_cache = cache._get_cache()
52+
53+
cache._clear()
54+
cache._save()
55+
comp_cache = cache._get_cache()
56+
57+
assert sorted(ref_cache) == sorted(comp_cache)

0 commit comments

Comments
 (0)