def _run_test_with(self, track_event, sensor_pos, torso_pos, expected_x, expected_y, expected_z): track_event.reset_mock() rp = RoomPosition() skeleton_3D = {} for j in rp.JOINTS: skeleton_3D[j] = torso_pos message = {'sensor_type': 'kinect', 'type': 'skeleton', 'sensor_position': sensor_pos, 'skeleton_3D': skeleton_3D, 'sensor_id': '10', 'created_at': 1, 'session_id': 'test:room-position'} #expected = {'sensor_id': '10', 'created_at': 1} expected = {} rp.process_message(message) eq_(track_event.call_count, 2) # one for subject_position and one for skeleton_in_room track_event.assert_any_call('test:room-position', 1, PositionMatcher( expected_x, expected_y, expected_z, expected))
def run_test_with(self, send_to, sensor_pos, torso_pos, expected_x, expected_y, expected_z, add_player_id = False): send_to.reset_mock() message = { 'sensor_type': 'kinect', 'type': 'skeleton', 'sensor_position': sensor_pos, 'skeleton_3D': {'torso': torso_pos}, 'sensor_id': '10', 'created_at': 1} expected = {'sensor_id': '10', 'created_at': 1} if add_player_id: message['player'] = 'xxx' expected['player'] = 'xxx' rp = RoomPosition() rp.process_message(message) eq_(send_to.call_count, 1) send_to.assert_any_call('subject-position',PositionMatcher(expected_x, expected_y, expected_z, expected))
def test_not_interesting_message(self, track_event): message = {'sensor_type': 'kinect', 'type': 'RGB'} rp = RoomPosition() rp.process_message(message) eq_(track_event.call_count, 0, "track_event should only be called if it has skeleton")
def test_not_interesting_message(self, send_to): message = { 'sensor_type': 'kinect', 'type': 'RGB'} rp = RoomPosition() rp.process_message(message) eq_(send_to.call_count, 0, "send_to should only be called if it has " " skeleton")