Mercurial > boomslang
view gamelib/tests/game_logic_utils.py @ 236:f82f017db104
Add bridge screenshot
author | Neil Muller <neil@dip.sun.ac.za> |
---|---|
date | Fri, 27 Aug 2010 09:06:06 +0200 |
parents | 30f2308c1efc |
children | 0bad554d0926 |
line wrap: on
line source
import unittest import pygame from gamelib import state from pygame.locals import SWSURFACE from gamelib.constants import SCREEN # We need this stuff set up so we can load images and whatnot. pygame.display.init() pygame.font.init() pygame.display.set_mode(SCREEN, SWSURFACE) class GameLogicTestCase(unittest.TestCase): CURRENT_SCENE = None def setUp(self): self.state = state.initial_state() self.state.set_current_scene(self.CURRENT_SCENE) def set_game_data(self, key, value, thing=None, scene=None): if scene is None: scene = self.CURRENT_SCENE gizmo = self.state.scenes[scene] if thing is not None: gizmo = gizmo.things[thing] gizmo.set_data(key, value) def assert_game_data(self, key, value, thing=None, scene=None): if scene is None: scene = self.CURRENT_SCENE gizmo = self.state.scenes[scene] if thing is not None: gizmo = gizmo.things[thing] self.assertEquals(value, gizmo.get_data(key)) def assert_inventory_item(self, item, in_inventory=True): self.assertEquals(in_inventory, self.state.items[item] in self.state.inventory) def assert_scene_thing(self, thing, in_scene=True): self.assertEquals(in_scene, thing in self.state.current_scene.things) def assert_detail_thing(self, thing, in_detail=True): self.assertEquals(in_detail, thing in self.state.current_detail.things) def interact_thing(self, thing, item=None, detail=False): item_obj = None if item is not None: item_obj = self.state.items[item] thing_container = self.state.current_scene if detail: thing_container = self.state.current_detail return thing_container.things[thing].interact(item_obj)