fix(tests): clear cache between tests

ssube · Mar 11, 2023 · 919f88c · 919f88c
1 parent db61697
commit 919f88c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/api/onnx_web/server/model_cache.py b/api/onnx_web/server/model_cache.py
@@ -53,6 +53,11 @@ def set(self, tag: str, key: Any, value: Any) -> None:
         cache.append((tag, key, value))
         self.prune()
 
+    def clear(self):
+        global cache
+
+        cache.clear()
+
     def prune(self):
         global cache
 

diff --git a/api/tests/server/test_model_cache.py b/api/tests/server/test_model_cache.py
@@ -5,25 +5,29 @@
 class TestStringMethods(unittest.TestCase):
   def test_drop_existing(self):
     cache = ModelCache(10)
+    cache.clear()
     cache.set("foo", ("bar",), {})
     self.assertGreater(cache.size, 0)
     self.assertEqual(cache.drop("foo", ("bar",)), 1)
 
   def test_drop_missing(self):
     cache = ModelCache(10)
+    cache.clear()
     cache.set("foo", ("bar",), {})
     self.assertGreater(cache.size, 0)
     self.assertEqual(cache.drop("foo", ("bin",)), 0)
 
   def test_get_existing(self):
     cache = ModelCache(10)
+    cache.clear()
     value = {}
     cache.set("foo", ("bar",), value)
     self.assertGreater(cache.size, 0)
     self.assertIs(cache.get("foo", ("bar",)), value)
 
   def test_get_missing(self):
     cache = ModelCache(10)
+    cache.clear()
     value = {}
     cache.set("foo", ("bar",), value)
     self.assertGreater(cache.size, 0)