def test_eq(self):
s = GameState()
s.ball.vel_x = 100
s.roles[0] = GameState.ROLE_BALL
s.paddle_left = GameObject()
s.paddle_left.pos_x = 99
s.players = [3,4,5]
t = GameState()
t.ball.vel_x = 100
t.roles[0] = GameState.ROLE_BALL
t.paddle_left = GameObject()
t.paddle_left.pos_x = 99
t.players = [3,4,5]
self.assertTrue(s == t)
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)
