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)
def setUp(self):
self._base_tracker_ut = BaseTracker()
