Mercurial > nagslang

changeset 125:12be9632fa15

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Better tests.
author Jeremy Thurgood <firxen@gmail.com>
date Mon, 02 Sep 2013 16:57:31 +0200
parents ae61528436f1
children c3af35561494
files nagslang/tests/test_game_object.py
diffstat 1 files changed, 9 insertions(+), 10 deletions(-) [+]
line wrap: on
line diff
--- a/nagslang/tests/test_game_object.py	Mon Sep 02 16:52:03 2013 +0200
+++ b/nagslang/tests/test_game_object.py	Mon Sep 02 16:57:31 2013 +0200
@@ -43,20 +43,19 @@
         puzzler.set_game_object(gobj)
         return puzzler
 
-    def test_floor_switch_puzzler(self):
-        gobj = FakeGameObject(None, FakeSpace())
+    def assert_floor_switch(self, expected, shapes):
+        gobj = FakeGameObject(None, FakeSpace(*shapes))
         puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
-        self.assertFalse(puzzler.get_state())
+        self.assertEqual(expected, puzzler.get_state())
 
-        gobj = FakeGameObject(None, FakeSpace(FakeShape()))
-        puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
-        self.assertFalse(puzzler.get_state())
+    def test_floor_switch_puzzler(self):
+        self.assert_floor_switch(False, [])
+        self.assert_floor_switch(False, [FakeShape()])
 
         for collision_type in SWITCH_PUSHERS:
-            gobj = FakeGameObject(
-                None, FakeSpace(FakeShape(collision_type)))
-            puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
-            self.assertTrue(puzzler.get_state())
+            self.assert_floor_switch(True, [FakeShape(collision_type)])
+            self.assert_floor_switch(
+                True, [FakeShape(), FakeShape(collision_type)])
 
     def test_state_proxy_puzzler(self):
         glue = game_object.PuzzleGlue()