Skip to content

Commit 919f88c

Browse files
committed
fix(tests): clear cache between tests
1 parent db61697 commit 919f88c

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

api/onnx_web/server/model_cache.py

+5
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,11 @@ def set(self, tag: str, key: Any, value: Any) -> None:
5353
cache.append((tag, key, value))
5454
self.prune()
5555

56+
def clear(self):
57+
global cache
58+
59+
cache.clear()
60+
5661
def prune(self):
5762
global cache
5863

api/tests/server/test_model_cache.py

+4
Original file line numberDiff line numberDiff line change
@@ -5,25 +5,29 @@
55
class TestStringMethods(unittest.TestCase):
66
def test_drop_existing(self):
77
cache = ModelCache(10)
8+
cache.clear()
89
cache.set("foo", ("bar",), {})
910
self.assertGreater(cache.size, 0)
1011
self.assertEqual(cache.drop("foo", ("bar",)), 1)
1112

1213
def test_drop_missing(self):
1314
cache = ModelCache(10)
15+
cache.clear()
1416
cache.set("foo", ("bar",), {})
1517
self.assertGreater(cache.size, 0)
1618
self.assertEqual(cache.drop("foo", ("bin",)), 0)
1719

1820
def test_get_existing(self):
1921
cache = ModelCache(10)
22+
cache.clear()
2023
value = {}
2124
cache.set("foo", ("bar",), value)
2225
self.assertGreater(cache.size, 0)
2326
self.assertIs(cache.get("foo", ("bar",)), value)
2427

2528
def test_get_missing(self):
2629
cache = ModelCache(10)
30+
cache.clear()
2731
value = {}
2832
cache.set("foo", ("bar",), value)
2933
self.assertGreater(cache.size, 0)

0 commit comments

Comments
 (0)