def testHitInHitsMatch(self): players = [Player('A'), Player('B'), Player('C')] classHits = HitsMatch(len(players), players) # Test usual bad hit for 0 player classHits._results_table = [ [1, 8, 2], [1, 8, 0], [None, None, None], ] classHits._current_player = 0 classHits._current_hole = 2 classHits.hit() self.assertEqual(classHits._results_table, [ [1, 8, 2], [1, 8, 0], [1, None, None], ]) for player in classHits._players_array: self.assertEqual(player.hits_no_more, 0) # Test usual good hit for 1 player classHits.hit(True) self.assertEqual(classHits._results_table, [ [1, 8, 2], [1, 8, 0], [1, 1, None], ]) self.assertEqual(classHits._players_array[0].hits_no_more, 0) self.assertEqual(classHits._players_array[1].hits_no_more, 1) self.assertEqual(classHits._players_array[2].hits_no_more, 0) # Test no hit for 9 points bad hit for 1 player => 10 points classHits._players_array[1].hits_no_more = 0 classHits._current_player = 1 classHits._current_hole = 0 classHits.hit() self.assertEqual(classHits._results_table, [ [1, 10, 2], [1, 8, 0], [1, 1, None], ]) self.assertEqual(classHits._players_array[1].hits_no_more, 1) # Test no hit for 8 points GOOD hit for 1 player => 9 points classHits._players_array[1].hits_no_more = 0 classHits._current_player = 1 classHits._current_hole = 1 classHits.hit(True) self.assertEqual(classHits._results_table, [ [1, 10, 2], [1, 9, 0], [1, 1, None], ]) self.assertEqual(classHits._players_array[1].hits_no_more, 1)
def testSetNextUser(self): players = [Player('A'), Player('B'), Player('C')] classHits = HitsMatch(len(players), players) # Should set next player 2(last), but not hole classHits._current_player = 1 classHits._current_hole = 1 classHits._set_next_player_and_hole_if_needed() self.assertEqual(classHits._current_hole, 1) self.assertEqual(classHits._current_player, 2) # Should set next player 0(first)!!!, but not hole classHits._set_next_player_and_hole_if_needed() self.assertEqual(classHits._current_hole, 1) self.assertEqual(classHits._current_player, 0) # Should set next player 2(= number of hole), AND next hole 2 for player in classHits._players_array: player.hits_no_more = 1 classHits._set_next_player_and_hole_if_needed() self.assertEqual(classHits._current_hole, 2) self.assertEqual(classHits._current_player, 2) # Should set finished state for player in classHits._players_array: player.hits_no_more = 1 classHits._set_next_player_and_hole_if_needed() self.assertEqual(classHits.finished, True)