def test_get_highest_speed_flag(self): dt = DataTransformer() r1 = 'Pikachu, Voltorb' r2 = 'Bulbasaur, Voltorb' winner = 'TIE' self.assertEqual(dt.get_highest_speed_flag(r1, r2), winner)