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"),
),
]
