test: update display test for segment1

tests/death_save.py

@@ -1,3 +1,5 @@
+#from objprint import op
+
 ##################
 # Helper Methods #
 ##################

tests/death_save_mc_testing.py

@@ -9,3 +9,18 @@ def _assertIncrement(self, var, switch, value):
 
     def _start(self):
         death_save._start(self)
+
+    # Helper function to check for existence
+    # of test on the display.
+    #
+    # Example:
+    #
+    #    self.assertIn('Some value',
+    #        [x.text for x in self._get_display_text_widgets])
+    #
+    def _get_display_text_widgets(self):
+        current_widgets = []
+        for w in self.mc.displays['apron'].current_slide.widgets:
+            if hasattr(w.widget, 'text'):
+                current_widgets.append(w.widget)
+        return current_widgets

tests/test_displays.py

@@ -2,44 +2,21 @@
 
 class TestDisplays(DeathSaveMcTesting):
 
-    def test_game_loop(self):
-        # TODO: Figure out how to test segment displays
-
+    def test_segment1(self):
         self.assertModeRunning("attract")
 
-        ## TODO: decide what to display on the segments
-        ##       during the attract mode
-        #widgets = self._get_display_text_widgets()
-        #self.assertIn(
-        #    "1234567", [x.text for x in widgets]
-        #)
-
         ## game begins and puts player1's score
         ## on the segment display
-        #self.hit_and_release_switch("s_start")
-        #self.advance_time_and_run(1)
-
-        #widgets = self._get_display_text_widgets()
-        #self.assertIn(
-        #    self.machine.game.player.score,
-        #    [int(x.text) for x in widgets]
-        #)
-
-        ## start game and player scores 10 points
-        ## by hitting the grand advance switch
-        #self.hit_and_release_switch("s_shooter_lane")
-        #self.hit_and_release_switch("s_grand_advance")
-        #self.advance_time_and_run(4)
-        #widgets = self._get_display_text_widgets()
-        #self.assertIn(10, [int(x.text) for x in widgets])
-
-    ##############
-    ### Helpers ##
-    ##############
-
-    #def _get_display_text_widgets(self):
-        #current_widgets = []
-        #for wgt in self.mc.displays['apron'].current_slide.widgets:
-        #    if hasattr(wgt.widget, 'text'):
-        #        current_widgets.append(wgt.widget)
-        #return current_widgets
+        self.hit_and_release_switch("s_start")
+        self.hit_and_release_switch("s_shooter_lane")
+        self.advance_time_and_run(1)
+
+        self.assertEqual(self.machine.game.player.score,
+            int(self.machine.segment_displays['segment1'].text))
+
+        ## player scores 10 points by hitting the
+        ## grand advance switch scoring 10 points
+        self.hit_and_release_switch("s_grand_advance")
+        self.advance_time_and_run(1)
+        self.assertEqual(10,
+            int(self.machine.segment_displays['segment1'].text))