Round.get_winner_summary added

2020-03-28 07:22:52 +01:00
parent ada5d74fd4
commit 07a494a40b
2 changed files with 48 additions and 0 deletions
@@ -275,6 +275,47 @@ class RoundCase(unittest.TestCase):
        r.stashes.append(Stash(2))
        self.assertEqual(r.get_last_stash(), r.stashes[1])

+    def test_get_winner_summary(self):
+        shuffler = lambda list: None
+        c0 = [
+                Card(Card_colors['BELLS'], Card_values['UPPER']),
+                Card(Card_colors['HEARTS'], Card_values['UPPER'])
+        ]
+        c1 = [
+                Card(Card_colors['BELLS'], Card_values['C7']),
+                Card(Card_colors['HEARTS'], Card_values['C10'])
+        ]
+        c2 = [
+                Card(Card_colors['BELLS'], Card_values['ACE']),
+                Card(Card_colors['BELLS'], Card_values['C8'])
+        ]
+        c3 = [
+                Card(Card_colors['LEAVES'], Card_values['C7']),
+                Card(Card_colors['BELLS'], Card_values['LOWER'])
+        ]
+        c = ['dummy']*24 + c0 + c1 + c2 + c3
+        r = Round(6, 1, c, shuffler)
+        self.assertEqual(r.get_winner_summary(), [0]*4)
+        r.add_player_guess(1, 0)
+        r.add_player_guess(2, 0)
+        r.add_player_guess(3, 1)
+        r.add_player_guess(0, 2)
+
+        r.play_card(0, c0[0])
+        r.play_card(1, c1[0])
+        r.play_card(2, c2[0])
+        self.assertEqual(r.get_winner_summary(), [0]*4)
+        r.play_card(3, c3[1])
+        self.assertEqual(r.get_winner_summary(), [0, 0, 1, 0])
+
+        r.play_card(2, c2[1])
+        r.play_card(3, c3[0])
+        r.play_card(0, c0[1])
+        r.play_card(1, c1[1])
+        self.assertEqual(r.get_winner_summary(), [1, 0, 1, 0])
+
+
+

 if __name__ == '__main__':
    unittest.main(verbosity=2)