def test_sends_debug_only_on_new_frame(self):
lwmm = LocalGoalModelModule()
return_value = lwmm.update({
DATA_KEY_IS_NEW_FRAME: False
})
self.assertEquals(-1, return_value)
def test_on_new_frame_publish_methods_are_called(self):
lwmm = LocalGoalModelModule()
publish_ball_info_mock = mock.MagicMock()
lwmm.publish_ball_info = publish_ball_info_mock
publish_goal_info_mock = mock.MagicMock()
lwmm.publish_goal_info = publish_goal_info_mock
publish_ball_info_prediction_mock = mock.MagicMock()
lwmm.publish_ball_info_prediction = publish_ball_info_prediction_mock
publish_obstacle_info_mock = mock.MagicMock()
lwmm.publish_obstacle_info = publish_obstacle_info_mock
data = {
DATA_KEY_IS_NEW_FRAME: True
}
lwmm.start(data)
return_value = None
for i in range(41):
return_value = lwmm.update(data)
publish_ball_info_mock.assert_called_once_with(data)
publish_goal_info_mock.assert_called_once_with(data)
publish_obstacle_info_mock.assert_called_once_with(data)
publish_ball_info_prediction_mock.assert_called_once_with(data)
self.assertEquals(None, return_value)
