mirror of https://github.com/m-labs/artiq
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
from artiq.experiment import *
|
|
from artiq.test.hardware_testbench import ExperimentCase
|
|
|
|
|
|
class _Cache(EnvExperiment):
|
|
def build(self):
|
|
self.setattr_device("core")
|
|
self.print = lambda x: print(x)
|
|
|
|
@kernel
|
|
def get(self, key):
|
|
return self.core.get_cache(key)
|
|
|
|
@kernel
|
|
def put(self, key, value):
|
|
self.core.put_cache(key, value)
|
|
|
|
@kernel
|
|
def get_put(self, key, value):
|
|
self.get(key)
|
|
self.put(key, value)
|
|
|
|
class CacheTest(ExperimentCase):
|
|
def test_get_empty(self):
|
|
exp = self.create(_Cache)
|
|
self.assertEqual(exp.get("x1"), [])
|
|
|
|
def test_put_get(self):
|
|
exp = self.create(_Cache)
|
|
exp.put("x2", [1, 2, 3])
|
|
self.assertEqual(exp.get("x2"), [1, 2, 3])
|
|
|
|
def test_replace(self):
|
|
exp = self.create(_Cache)
|
|
exp.put("x3", [1, 2, 3])
|
|
exp.put("x3", [1, 2, 3, 4, 5])
|
|
self.assertEqual(exp.get("x3"), [1, 2, 3, 4, 5])
|
|
|
|
def test_borrow(self):
|
|
exp = self.create(_Cache)
|
|
exp.put("x4", [1, 2, 3])
|
|
with self.assertRaises(CacheError):
|
|
exp.get_put("x4", [])
|