def test_indicator_frame_set(self, indicator_mock): indicator_mock.ref = "test_indicator_ref" fs = FrameSetFactory() fs.add_indicator(indicator_mock) assert indicator_mock.set_frame_set.call_args_list == [call(fs)]
def test_set_as_indicator(self, indicator_mock): indicator_mock.ref = "test_indicator_ref" fs = FrameSetFactory() fs.add_indicator(indicator_mock) assert fs.indicators == {"test_indicator_ref": indicator_mock}
def test_indicators_not_empty(self, mocker): epic_mock = mocker.Mock() epic_mock.market_open = True parent_framset = FrameSetFactory() indicator_mock = mocker.Mock() indicator_mock.ref = "my_indicator_ref" indicator_mock.build_value_from_frame.return_value = "indicator_value" parent_framset.epic = epic_mock parent_framset.add_indicator(indicator_mock) fr = FrameFactory(parent_frameset=parent_framset) assert fr.indicators == {"my_indicator_ref": "indicator_value"}
def test_indicators_on_new_tick_called__indicator_none( self, mocker, market_open): epic_mock = mocker.Mock() epic_mock.market_open = market_open parent_framset = FrameSetFactory() indicator_mock = mocker.Mock() indicator_mock.ref = "my_indicator_ref" parent_framset.epic = epic_mock parent_framset.add_indicator(indicator_mock) fr = FrameFactory(parent_frameset=parent_framset) fr.indicators = {"my_indicator_ref": None} tick = TickFactory() fr.on_new_tick(tick) assert indicator_mock.build_value_from_frame.call_args_list == [ call(fr, market_open), # call on Frame Init call(fr, market_open), ]