def test_initially_moving_to_middle(self): zone = self.world_left.pitch.zones[2][0] target_x, target_y = self.strategy_left._get_shooting_coordinates( self.world_left.our_attacker) for x, y in zone: robot = self.get_aimed_robot( x, y, target_x, target_y, self.world_left.our_attacker.zone) world = World('left') self.place_robot(world, world.our_attacker.zone, robot) strategy = AttackerScoreDynamic(world) actions = strategy.generate() left_motor = actions['left_motor'] right_motor = actions['right_motor'] # Both left motor and right motor should be positive or negative. self.assertEqual(True, left_motor * right_motor > 0)
def test_confuse1_rotation_up(self): target_x, target_y = self.strategy_left.shooting_pos aim_x = self.world_left.their_goal.x aim_y = self.world_left.their_goal.y + self.world_left.their_goal.height / 2 + self.strategy_left.GOAL_CORNER_OFFSET attacker = self.get_aimed_robot( target_x, target_y, aim_x, aim_y, self.world_left.our_attacker.zone) defender = self.get_aimed_robot( self.world_left.their_goal.x, self.world_left.their_goal.y, aim_x, aim_y, self.world_left.their_defender.zone) world = World('left') self.place_robot(world, attacker.zone, attacker) self.place_robot(world, defender.zone, defender) strategy = AttackerScoreDynamic(world) actions = strategy.generate() self.assertTrue(actions['left_motor'] < 0) self.assertTrue(actions['right_motor'] > 0)
def setUp(self): self.world_left = World('left') self.strategy_left = AttackerScoreDynamic(self.world_left) self.world_right = World('right') self.strategy_right = AttackerScoreDynamic(self.world_right)
def setUp(self): self.world_left = World('left') self.strategy_left = DefaultDefenderDefence(self.world_left)