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