def test_add_robot(self): conf = TestRobotState.default_config() state = RobotState(conf) sidebar = Sidebar(100, 150, 200, 300) sidebar_sprite_mock = MagicMock() sidebar.init_robot(state.name, state.sensors, state.get_bricks(), [sidebar_sprite_mock]) self.assertEqual(len(sidebar.robot_info), 1)
def test_setters_getters(self): state = RobotState(self.default_config()) state.setup_pymunk_shapes(1) self.assertEqual(len(state.get_shapes()), 4 + 1 + 2 + 1) self.assertEqual( state.get_sensor((0, 'ev3-ports:in4')).get_ev3type(), 'ultrasonic_sensor') self.assertEqual(state.get_bricks()[0].name, 'brick-left') self.assertEqual(len(state.get_sensors()), 1) self.assertEqual(state.get_anchor().name, 'brick-left') del state.bricks[0] self.assertEqual(state.get_anchor(), None)
def test_add_robot_info(self): conf = TestRobotState.default_config() state = RobotState(conf) sidebar = Sidebar(100, 150, 200, 300) sidebar_sprite_mock = MagicMock() state.values[(0, 'ev3-ports:in4')] = 5 state.sounds[(0, 'speaker')] = 'test_sound' sidebar.init_robot(state.name, state.sensors, state.get_bricks(), [sidebar_sprite_mock]) sidebar.add_robot_info(state.name, state.values, state.sounds) self.assertEqual( sidebar.robot_info[state.name][(0, 'ev3-ports:in4')]['value'], 5) self.assertEqual( sidebar.robot_info[state.name][(0, 'speaker')]['value'], 'test_sound')