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]