2023-03-06 13:34:09 +00:00
|
|
|
import unittest
|
|
|
|
|
|
|
|
from onnx_web.server.model_cache import ModelCache
|
|
|
|
|
2023-09-13 22:27:44 +00:00
|
|
|
|
|
|
|
class TestModelCache(unittest.TestCase):
|
2023-11-20 05:18:57 +00:00
|
|
|
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)
|
|
|
|
self.assertIs(cache.get("foo", ("bin",)), None)
|
|
|
|
|
|
|
|
"""
|
2023-09-13 22:27:44 +00:00
|
|
|
def test_set_existing(self):
|
|
|
|
cache = ModelCache(10)
|
|
|
|
cache.clear()
|
|
|
|
cache.set("foo", ("bar",), {
|
|
|
|
"value": 1,
|
|
|
|
})
|
|
|
|
value = {
|
|
|
|
"value": 2,
|
|
|
|
}
|
|
|
|
cache.set("foo", ("bar",), value)
|
|
|
|
self.assertIs(cache.get("foo", ("bar",)), value)
|
|
|
|
"""
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
def test_set_missing(self):
|
|
|
|
cache = ModelCache(10)
|
|
|
|
cache.clear()
|
|
|
|
value = {}
|
|
|
|
cache.set("foo", ("bar",), value)
|
|
|
|
self.assertIs(cache.get("foo", ("bar",)), value)
|
|
|
|
|
|
|
|
def test_set_zero(self):
|
|
|
|
cache = ModelCache(0)
|
|
|
|
cache.clear()
|
|
|
|
value = {}
|
|
|
|
cache.set("foo", ("bar",), value)
|
|
|
|
self.assertEqual(cache.size, 0)
|