Mercurial > sypikslang
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())