Mercurial > pyntnclick
annotate gamelib/tests/game_logic_utils.py @ 137:17411b984dc2
Fix descriptions. And call Demeter's police down upon me.
author | Jeremy Thurgood <firxen@gmail.com> |
---|---|
date | Tue, 24 Aug 2010 20:51:40 +0200 |
parents | 6b0f7364f3bf |
children | 30f2308c1efc |
rev | line source |
---|---|
60 | 1 import unittest |
2 | |
3 import pygame | |
4 | |
5 from gamelib import state | |
6 | |
7 from pygame.locals import SWSURFACE | |
8 from gamelib.constants import SCREEN | |
9 | |
10 | |
11 # We need this stuff set up so we can load images and whatnot. | |
12 pygame.display.init() | |
13 pygame.display.set_mode(SCREEN, SWSURFACE) | |
14 | |
15 | |
16 class GameLogicTestCase(unittest.TestCase): | |
17 CURRENT_SCENE = None | |
18 | |
19 def setUp(self): | |
20 self.state = state.initial_state() | |
21 self.state.set_current_scene(self.CURRENT_SCENE) | |
22 | |
23 def set_game_data(self, key, value, thing=None, scene=None): | |
24 if scene is None: | |
25 scene = self.CURRENT_SCENE | |
26 gizmo = self.state.scenes[scene] | |
27 if thing is not None: | |
28 gizmo = gizmo.things[thing] | |
29 gizmo.set_data(key, value) | |
30 | |
31 def assert_game_data(self, key, value, thing=None, scene=None): | |
32 if scene is None: | |
33 scene = self.CURRENT_SCENE | |
34 gizmo = self.state.scenes[scene] | |
35 if thing is not None: | |
36 gizmo = gizmo.things[thing] | |
37 self.assertEquals(value, gizmo.get_data(key)) | |
38 | |
67
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
39 def assert_inventory_item(self, item, in_inventory=True): |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
40 self.assertEquals(in_inventory, self.state.items[item] in self.state.inventory) |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
41 |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
42 def interact_thing(self, thing, item=None): |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
43 item_obj = None |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
44 if item is not None: |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
45 item_obj = self.state.items[item] |
6b0f7364f3bf
Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents:
60
diff
changeset
|
46 self.state.scenes[self.CURRENT_SCENE].things[thing].interact(item_obj) |