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")
