def test_pick_first_with_nonexistent_team(self): e = None player1 = Player("Boyan", "Team1") player2 = Player("Silvia", "Team1") player3 = Player("Marto", "Team2") player4 = Player("Gosho", "Team2") players = [[player1, player2], [player3, player4]] try: g = Belote(players) g.pick_first_from("Team3") except ValueError as exc: e = exc self.assertIsNotNone(e) self.assertEqual(str(e), "Team not found.")
def test_pick_first(self): e = None player1 = Player("Boyan", "Team1") player2 = Player("Silvia", "Team1") player3 = Player("Marto", "Team2") player4 = Player("Gosho", "Team2") players = [[player1, player2], [player3, player4]] try: g = Belote(players) g.pick_first_from("Team2") first = g.__dict__["order"][0] first = g.__dict__["players"][first] third = g.__dict__["order"][2] third = g.__dict__["players"][third] except Exception as exc: e = exc self.assertIsNone(e) self.assertEqual(first.team, "Team2") self.assertEqual(third.team, "Team2")