def test_apply_frames_limit(self, mocker):
fs = FrameSetFactory(max_frames_in_memory=3)
frame_mock = mocker.Mock()
frame_mock.previous_frame = "test"
for frame_idx in range(10):
fs.frames.append(frame_mock)
fs.clean_old_frames()
assert len(fs.frames) == 3
def test_keep_only_last_frames(self, mocker):
fs = FrameSetFactory(max_frames_in_memory=3)
for frame_idx in range(10):
frame_mock = mocker.Mock()
frame_mock.previous_frame = "test"
frame_mock.idx = frame_idx
fs.frames.append(frame_mock)
fs.clean_old_frames()
assert [frame.idx for frame in fs.frames] == [7, 8, 9]
