fix test case when dining hall is closed

pennlabs · Mar 7, 2018 · cb8c6b2 · cb8c6b2
1 parent 16be358
commit cb8c6b2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/dining_test.py b/tests/dining_test.py
@@ -18,7 +18,9 @@ def test_dining(self):
         self.assertTrue(len(venues) > 0)
         id = str(venues["id"])
         data = self.din.menu_daily(id)
-        self.assertTrue(len(data["result_data"]["Document"]["tblMenu"]["tblDayPart"][0]) >= 2)
+        dayPart = data["result_data"]["Document"]["tblMenu"]["tblDayPart"]
+        if len(dayPart) > 0:
+            self.assertTrue(len(dayPart[0]) >= 2)
 
     def test_dining_normalization(self):
         data = self.din.menu_daily("593")