def test_make_jitcell_getter_custom(): from pypy.rpython.typesystem import LowLevelTypeSystem class FakeRTyper: type_system = LowLevelTypeSystem.instance class FakeJitCell(BaseJitCell): pass celldict = {} def getter(x, y): return celldict.get((x, y)) def setter(newcell, x, y): newcell.x = x newcell.y = y celldict[x, y] = newcell GETTER = lltype.Ptr(lltype.FuncType([lltype.Signed, lltype.Float], llmemory.GCREF)) SETTER = lltype.Ptr(lltype.FuncType([llmemory.GCREF, lltype.Signed, lltype.Float], lltype.Void)) class FakeWarmRunnerDesc: rtyper = FakeRTyper() cpu = None get_jitcell_at_ptr = llhelper(GETTER, getter) set_jitcell_at_ptr = llhelper(SETTER, setter) # state = WarmEnterState(FakeWarmRunnerDesc()) get_jitcell = state._make_jitcell_getter_custom(FakeJitCell) cell1 = get_jitcell(5, 42.5) assert isinstance(cell1, FakeJitCell) assert cell1.x == 5 assert cell1.y == 42.5 cell2 = get_jitcell(5, 42.5) assert cell2 is cell1 cell3 = get_jitcell(41, 42.5) cell4 = get_jitcell(42, 0.25) assert cell1 is not cell3 is not cell4 is not cell1
def test_make_jitcell_getter_custom(): from pypy.rpython.typesystem import LowLevelTypeSystem class FakeRTyper: type_system = LowLevelTypeSystem.instance celldict = {} def getter(x, y): return celldict.get((x, y)) def setter(newcell, x, y): newcell.x = x newcell.y = y celldict[x, y] = newcell GETTER = lltype.Ptr( lltype.FuncType([lltype.Signed, lltype.Float], llmemory.GCREF)) SETTER = lltype.Ptr( lltype.FuncType([llmemory.GCREF, lltype.Signed, lltype.Float], lltype.Void)) class FakeWarmRunnerDesc: rtyper = FakeRTyper() cpu = None memory_manager = None class FakeJitDriverSD: _get_jitcell_at_ptr = llhelper(GETTER, getter) _set_jitcell_at_ptr = llhelper(SETTER, setter) # state = WarmEnterState(FakeWarmRunnerDesc(), FakeJitDriverSD()) get_jitcell = state._make_jitcell_getter_custom() cell1 = get_jitcell(True, 5, 42.5) assert isinstance(cell1, JitCell) assert cell1.x == 5 assert cell1.y == 42.5 cell2 = get_jitcell(True, 5, 42.5) assert cell2 is cell1 cell3 = get_jitcell(True, 41, 42.5) assert get_jitcell(False, 42, 0.25) is None cell4 = get_jitcell(True, 42, 0.25) assert get_jitcell(False, 42, 0.25) is cell4 assert cell1 is not cell3 is not cell4 is not cell1