diff gamelib/tests/test_lab.py @ 151:372d886f9e70

New suggest_research() method on Lab.
author Jeremy Thurgood <firxen@gmail.com>
date Fri, 11 May 2012 20:06:36 +0200
parents 3abb05f7e720
children 5b58e2150a4f
line wrap: on
line diff
--- a/gamelib/tests/test_lab.py	Fri May 11 17:17:44 2012 +0200
+++ b/gamelib/tests/test_lab.py	Fri May 11 20:06:36 2012 +0200
@@ -1,6 +1,6 @@
 from unittest import TestCase
 
-from gamelib.lab import Lab
+from gamelib.lab import Lab, ScienceGraph
 from gamelib import research, schematics
 
 
@@ -9,6 +9,7 @@
         'research.Physics': 5,
         'research.Rocketry': 2,
         'research.Electrickery': 3,
+        'schematic.MachineGun': 1,
         },
     }
 
@@ -46,3 +47,8 @@
         new_science = dict(
             LAB_DATA['science'].items() + [('schematic.LightningGun', 0)])
         self.assertEqual({'science': new_science}, lab.save_data())
+
+    def test_science_graph(self):
+        lab = Lab(LAB_DATA)
+        self.assertEqual((True, [research.Electrickery, research.Physics]),
+                         lab.suggest_research())