In this problem, you will work with the Atari game Pong, a classic game environment. Your goal is to play the game by applying actions to the game state. The environment returns the state of the game and you need to decide the action to take (up, down, left, right) to hit the ball. **Example:** Suppose the current state is [[0.1, 0.2], [0.3, 0.4]] representing the paddle position and ball position. You need to decide whether to move the paddle up (0), down (1), left (2), or right (3). **Constraints:** The paddle position should be within [0, 1] and the ball position should be within [0, 1].
Test Cases
Test Case 1
Input:
[[0.5, 0.6], [0.7, 0.8]]Expected:
[[0.49, 0.6], [0.7, 0.8]]Test Case 2
Input:
[[0.1, 0.2], [0.3, 0.4]]Expected:
[[0.1, 0.2], [0.3, 0.4]]+ 3 hidden test cases