Add tests for game.py