fix(tests): clear cache between tests
This commit is contained in:
parent
db6169705b
commit
919f88c7d0
|
@ -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
|
||||||
|
|
||||||
|
|
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue