Skip to content

Commit 1c12003

Browse files
committed
tests: Add IR cache clearing test
1 parent 640bda9 commit 1c12003

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

tests/test_ir.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1145,3 +1145,22 @@ def __init__(self, value: int):
11451145
hits, _, _, cursize = Object._instance_cache.cache_info()
11461146
assert hits == 1 # obj5 hit the cache
11471147
assert cursize == 2
1148+
1149+
def test_cleared_after_build(self):
1150+
"""
1151+
Tests that instance caches are cleared after building an Operator.
1152+
"""
1153+
class Object(CacheInstances):
1154+
def __init__(self, value: int):
1155+
self.value = value
1156+
1157+
obj1 = Object(1)
1158+
cache_size = Object._instance_cache.cache_info()[-1]
1159+
assert cache_size == 1
1160+
1161+
x = Symbol('x')
1162+
Operator(Eq(x, obj1.value))
1163+
1164+
# Cache should be cleared after Operator construction
1165+
cache_size = Object._instance_cache.cache_info()[-1]
1166+
assert cache_size == 0

0 commit comments

Comments
 (0)