Mercurial > sypikslang
diff gamelib/tests/test_lab.py @ 17:10d3db1f1e08
Set up initial research and rework breakthroughs.
| author | Jeremy Thurgood <firxen@gmail.com> |
|---|---|
| date | Sun, 06 May 2012 17:39:37 +0200 |
| parents | 9d61abb3cfaf |
| children | 718d1ec382f7 |
line wrap: on
line diff
--- a/gamelib/tests/test_lab.py Sun May 06 17:12:40 2012 +0200 +++ b/gamelib/tests/test_lab.py Sun May 06 17:39:37 2012 +0200 @@ -9,11 +9,23 @@ self.lab = Lab() def test_find_new_products(self): + # Set up the appropriate research. + if research.Tesla in self.lab.new_research: + self.lab._gain_science(research.Tesla(1)) + self.lab._get_science(research.Tesla).points += 2 + # Check breakthrough options. new_products = self.lab.find_new_products(None) - self.assertTrue(products.MachineGun in new_products) + self.assertTrue(products.TeslaTank in new_products) self.assertTrue(products.DoomsdayVirus not in new_products) def test_find_new_research(self): - new_research = self.lab.find_new_research(None) - self.assertTrue(research.Tesla in new_research) - self.assertTrue(research.Space not in new_research) + # Set up the appropriate research. + if research.Robotics in self.lab.new_research: + self.lab._gain_science(research.Robotics(1)) + if research.Rocketry in self.lab.new_research: + self.lab._gain_science(research.Rocketry(1)) + self.lab._get_science(research.Rocketry).points += 1 + # Check breakthrough options. + new_research = self.lab.find_new_research() + self.assertTrue(research.Space in new_research) + self.assertTrue(research.ArtificialIntelligence not in new_research)