def test_is_frame_over_tick_call(self, mock_is_frame_over_tick): fs = FrameSetFactory(unit=Unit.TICK) tick = TickFactory() fs.on_new_tick(tick) assert mock_is_frame_over_tick.call_args_list == [call(tick)]
def test_add_to_last_tick(self, mocker, mock_is_frame_over_timed): mock_is_frame_over_timed.return_value = False add_tick_to_last_mock = mocker.patch( f"{CLASS_DEFINITION_PATH}.add_tick_to_last", ) fs = FrameSetFactory(unit=Unit.SECOND) tick = TickFactory() fs.on_new_tick(tick) assert add_tick_to_last_mock.call_args_list == [call(tick)]
def test_add_tick_to_last_call(self, mocker, mock_is_frame_over_tick): mock_is_frame_over_tick.return_value = False fs = FrameSetFactory(unit=Unit.TICK) add_tick_to_last_mock = mocker.patch.object(fs, "add_tick_to_last", wraps=fs.add_tick_to_last) tick = TickFactory() fs.on_new_tick(tick) assert add_tick_to_last_mock.call_args_list == [call(tick)]
def test_create_new_frame_call__no_gap(self, mocker): create_new_frame_mock = mocker.patch( f"{CLASS_DEFINITION_PATH}.create_new_frame", ) fs = FrameSetFactory(unit=Unit.SECOND, unit_quantity=10) tick = TickFactory(datetime=arrow.get("2020-01-01 12:34:56")) fs.on_new_tick(tick) assert create_new_frame_mock.call_args_list == [ call(tick, arrow.get("2020-01-01 12:34:50"), arrow.get("2020-01-01 12:35:00")) ]
def test_create_new_frame_call(self, mocker, mock_is_frame_over_tick): mock_is_frame_over_tick.return_value = True fs = FrameSetFactory(unit=Unit.TICK) create_new_frame_mock = mocker.patch.object(fs, "create_new_frame", wraps=fs.create_new_frame) tick = TickFactory() fs.on_new_tick(tick) assert create_new_frame_mock.call_args_list == [ call(tick=tick, period_start=tick.datetime, period_end=None) ]
def test_create_new_frame_call__gap(self, mocker): fs = FrameSetFactory(unit=Unit.SECOND, unit_quantity=10) create_new_frame_mock = mocker.patch.object(fs, "create_new_frame", wraps=fs.create_new_frame) current_last_tick = TickFactory() current_frame = FrameFactory( first_tick=current_last_tick, period_start=arrow.get("2020-01-01 12:34:10"), period_end=arrow.get("2020-01-01 12:34:20"), ) fs.frames = [current_frame] new_tick = TickFactory(datetime=arrow.get("2020-01-01 12:34:56")) fs.on_new_tick(new_tick) assert create_new_frame_mock.call_args_list == [ call( current_last_tick, arrow.get("2020-01-01 12:34:20"), arrow.get("2020-01-01 12:34:30"), ), call( current_last_tick, arrow.get("2020-01-01 12:34:30"), arrow.get("2020-01-01 12:34:40"), ), call( current_last_tick, arrow.get("2020-01-01 12:34:40"), arrow.get("2020-01-01 12:34:50"), ), call( new_tick, arrow.get("2020-01-01 12:34:50"), arrow.get("2020-01-01 12:35:00"), ), ]