apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(tests): clear cache between tests

This commit is contained in:
Sean Sube 2023-03-11 13:59:09 -06:00
parent db6169705b
commit 919f88c7d0
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/server/model_cache.py
View File

@ -53,6 +53,11 @@ class ModelCache:
cache.append((tag, key, value)) cache.append((tag, key, value))
self.prune() self.prune()
def clear(self):
global cache
cache.clear()
def prune(self): def prune(self):
global cache global cache

4
api/tests/server/test_model_cache.py
View File

@ -5,18 +5,21 @@ from onnx_web.server.model_cache import ModelCache
class TestStringMethods(unittest.TestCase): class TestStringMethods(unittest.TestCase):
def test_drop_existing(self): def test_drop_existing(self):
cache = ModelCache(10) cache = ModelCache(10)
cache.clear()
cache.set("foo", ("bar",), {}) cache.set("foo", ("bar",), {})
self.assertGreater(cache.size, 0) self.assertGreater(cache.size, 0)
self.assertEqual(cache.drop("foo", ("bar",)), 1) self.assertEqual(cache.drop("foo", ("bar",)), 1)
def test_drop_missing(self): def test_drop_missing(self):
cache = ModelCache(10) cache = ModelCache(10)
cache.clear()
cache.set("foo", ("bar",), {}) cache.set("foo", ("bar",), {})
self.assertGreater(cache.size, 0) self.assertGreater(cache.size, 0)
self.assertEqual(cache.drop("foo", ("bin",)), 0) self.assertEqual(cache.drop("foo", ("bin",)), 0)
def test_get_existing(self): def test_get_existing(self):
cache = ModelCache(10) cache = ModelCache(10)
cache.clear()
value = {} value = {}
cache.set("foo", ("bar",), value) cache.set("foo", ("bar",), value)
self.assertGreater(cache.size, 0) self.assertGreater(cache.size, 0)
@ -24,6 +27,7 @@ class TestStringMethods(unittest.TestCase):
def test_get_missing(self): def test_get_missing(self):
cache = ModelCache(10) cache = ModelCache(10)
cache.clear()
value = {} value = {}
cache.set("foo", ("bar",), value) cache.set("foo", ("bar",), value)
self.assertGreater(cache.size, 0) self.assertGreater(cache.size, 0)