def test_update_scoring_init(self): b = BaseTracker(tracked_ids=['red', 'blue']) b.update_scoring() self.assertEqual(b.score['red'], 0) self.assertEqual(b.score['blue'], 0) self.assertEqual(b.flag['red'], False) self.assertEqual(b.flag['blue'], False)
def test_update_scoring_get_blue_gets_flag(self): b = BaseTracker(tracked_ids=['red', 'blue']) b.center['blue'] = constants.BASE['red'] b.update_scoring() self.assertEqual(b.score['red'], 0) self.assertEqual(b.score['blue'], 0) self.assertEqual(b.flag['red'], False) self.assertEqual(b.flag['blue'], True)
class TestBaseTracker(unittest.TestCase): def setUp(self): self._base_tracker_ut = BaseTracker() def test_name(self): self.assertEquals('BaseTracker', self._base_tracker_ut.name) def test_extract_url(self): self.assertEquals('test', self._base_tracker_ut.extract_download_url('test'))
def setUp(self): self._base_tracker_ut = BaseTracker()