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)