# HG changeset patch
# User Jeremy Thurgood <firxen@gmail.com>
# Date 1378133851 -7200
# Node ID 12be9632fa15a57c9e059296d22f904c94dbb817
# Parent  ae61528436f1da848102e4c81dbbe1e56562b823
Better tests.

diff -r ae61528436f1 -r 12be9632fa15 nagslang/tests/test_game_object.py
--- 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()