def test_roll(self): player = Player('red', 'fakeid') player.dice = 2 player.white_dice = 0 fakeroll = iter([2, 2]) player.roll(randint=fakeroll.next) self.assertEqual(player.rolled_dice, {2: [2, 0]}) player.dice = 3 player.white_dice = 1 fakeroll = iter([2, 2, 1, 1]) player.roll(randint=fakeroll.next) self.assertEqual(player.rolled_dice, {2: [2, 0], 1: [1, 1]})
def test_remaining_dice(self): player = Player('red', 'fakeid') player.dice = 1 player.white_dice = 1 self.assertEqual(player.remaining_dice(), 2) player.dice = 0 self.assertEqual(player.remaining_dice(), 1)