def test_ResumeDataLoopMemo_number_boxes():
memo = ResumeDataLoopMemo(FakeMetaInterpStaticData())
b1, b2 = [IntFrontendOp(0), IntFrontendOp(0)]
assert memo.num_cached_boxes() == 0
boxes = []
num = memo.assign_number_to_box(b1, boxes)
assert num == -1
assert boxes == [b1]
assert memo.num_cached_boxes() == 1
boxes = [None]
num = memo.assign_number_to_box(b1, boxes)
assert num == -1
assert boxes == [b1]
num = memo.assign_number_to_box(b2, boxes)
assert num == -2
assert boxes == [b1, b2]
assert memo.num_cached_boxes() == 2
boxes = [None, None]
num = memo.assign_number_to_box(b2, boxes)
assert num == -2
assert boxes == [None, b2]
num = memo.assign_number_to_box(b1, boxes)
assert num == -1
assert boxes == [b1, b2]
memo.clear_box_virtual_numbers()
assert memo.num_cached_boxes() == 0
def test_ResumeDataLoopMemo_number_virtuals():
memo = ResumeDataLoopMemo(FakeMetaInterpStaticData())
b1, b2 = [IntFrontendOp(0), IntFrontendOp(0)]
assert memo.num_cached_virtuals() == 0
num = memo.assign_number_to_virtual(b1)
assert num == -1
assert memo.num_cached_virtuals() == 1
num = memo.assign_number_to_virtual(b1)
assert num == -1
num = memo.assign_number_to_virtual(b2)
assert num == -2
assert memo.num_cached_virtuals() == 2
num = memo.assign_number_to_virtual(b2)
assert num == -2
num = memo.assign_number_to_virtual(b1)
assert num == -1
memo.clear_box_virtual_numbers()
assert memo.num_cached_virtuals() == 0