class RobotTests(object): def setup_helper(self): self.conn = MagicMock() self.sensor_state = defaultdict(int) self.sense = MorseSense(self.conn, self.sensor_state) self.conn.subscribe = MagicMock( side_effect=self.subscribe_side_effect() ) self.conn.unscribe = MagicMock() self.sense.start(.001) def assert_bot_detection(self, robot): self.assertEqual(self.sensor_state['robot'], robot) def assert_sensor_data(self, decoded_sensor): for key, value in decoded_sensor.iteritems(): self.assertEqual(self.sensor_state[key], value) def assert_start(self): self.assertEqual(self.conn.subscribe.call_count, 2) def assert_stop(self): self.assertEqual(self.conn.subscribe.call_count, 2) self.conn.reset_mock() self.sense.unsubscribe() self.assertEqual(self.conn.unsubscribe.call_count, 2)
def setup_helper(self): self.conn = MagicMock() self.sensor_state = defaultdict(int) self.sense = MorseSense(self.conn, self.sensor_state) self.conn.subscribe = MagicMock( side_effect=self.subscribe_side_effect()) self.conn.unscribe = MagicMock() self.sense.start(.001)
def setup_helper(self): self.conn = MagicMock() self.sensor_state = defaultdict(int) self.sense = MorseSense(self.conn, self.sensor_state) self.conn.subscribe = MagicMock( side_effect=self.subscribe_side_effect() ) self.conn.unscribe = MagicMock() self.sense.start(.001)