Changeset 125:12be9632fa15

Timestamp:

Sep 2, 2013, 2:57:31 PM (7 years ago)

Author:

Jeremy Thurgood <firxen@…>

Branch:

default

Message:

Better tests.

File:

• nagslang/tests/test_game_object.py (modified) (1 diff)

nagslang/tests/test_game_object.py

@@ -44 +44 @@
         return puzzler
 
+    def assert_floor_switch(self, expected, shapes):
+        gobj = FakeGameObject(None, FakeSpace(*shapes))
+        puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
+        self.assertEqual(expected, puzzler.get_state())
+
     def test_floor_switch_puzzler(self):
-        gobj = FakeGameObject(None, FakeSpace())
-        puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
-        self.assertFalse(puzzler.get_state())
-
-        gobj = FakeGameObject(None, FakeSpace(FakeShape()))
-        puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
-        self.assertFalse(puzzler.get_state())
+        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):