def test_RotateRoles_1(self): e = UDPGameEngine() s = GameState() s.player_size = 1 s.roles = [GameState.ROLE_LEFT_PADDLE] s.players = [-1, 0] e.RotateRoles(s) self.assertTrue(s.roles == [GameState.ROLE_LEFT_PADDLE]) self.assertTrue(s.players == [-1, 0])
def test_RotateRoles_2(self): e = UDPGameEngine() s = GameState() s.player_size = 2 s.roles = [GameState.ROLE_LEFT_PADDLE, GameState.ROLE_RIGHT_PADDLE] s.players = [-1]*(s.player_size + 1) s.players[GameState.ROLE_LEFT_PADDLE] = 0 s.players[GameState.ROLE_RIGHT_PADDLE] = 1 e.RotateRoles(s) self.assertTrue(s.roles == [GameState.ROLE_RIGHT_PADDLE, GameState.ROLE_LEFT_PADDLE]) self.assertTrue(s.players == [-1, 1, 0])
def test_RotateRoles_3(self): e = UDPGameEngine() s = GameState() s.roles = [GameState.ROLE_RIGHT_PADDLE, GameState.ROLE_LEFT_PADDLE, GameState.ROLE_BALL] s.player_size = len(s.roles) s.players = [-1]*(len(s.roles)+ 1) s.players[GameState.ROLE_LEFT_PADDLE] = 1 s.players[GameState.ROLE_RIGHT_PADDLE] = 0 s.players[GameState.ROLE_BALL] = 2 e.RotateRoles(s) self.assertTrue(s.roles == [GameState.ROLE_LEFT_PADDLE, GameState.ROLE_BALL, GameState.ROLE_RIGHT_PADDLE]) players = [-1]*(len(s.roles)+1) players[GameState.ROLE_LEFT_PADDLE] = 0 players[GameState.ROLE_RIGHT_PADDLE] = 2 players[GameState.ROLE_BALL] = 1 self.assertTrue(s.players == players)