Mercurial > sypikslang

comparison gamelib/tests/test_lab.py @ 17:10d3db1f1e08

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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
comparison
equal deleted inserted replaced
16:af1bfeb648cb 17:10d3db1f1e08
7 class TestLab(TestCase): 7 class TestLab(TestCase):
8 def setUp(self): 8 def setUp(self):
9 self.lab = Lab() 9 self.lab = Lab()
10 10
11 def test_find_new_products(self): 11 def test_find_new_products(self):
12 # Set up the appropriate research.
13 if research.Tesla in self.lab.new_research:
14 self.lab._gain_science(research.Tesla(1))
15 self.lab._get_science(research.Tesla).points += 2
16 # Check breakthrough options.
12 new_products = self.lab.find_new_products(None) 17 new_products = self.lab.find_new_products(None)
13 self.assertTrue(products.MachineGun in new_products) 18 self.assertTrue(products.TeslaTank in new_products)
14 self.assertTrue(products.DoomsdayVirus not in new_products) 19 self.assertTrue(products.DoomsdayVirus not in new_products)
15 20
16 def test_find_new_research(self): 21 def test_find_new_research(self):
17 new_research = self.lab.find_new_research(None) 22 # Set up the appropriate research.
18 self.assertTrue(research.Tesla in new_research) 23 if research.Robotics in self.lab.new_research:
19 self.assertTrue(research.Space not in new_research) 24 self.lab._gain_science(research.Robotics(1))
25 if research.Rocketry in self.lab.new_research:
26 self.lab._gain_science(research.Rocketry(1))
27 self.lab._get_science(research.Rocketry).points += 1
28 # Check breakthrough options.
29 new_research = self.lab.find_new_research()
30 self.assertTrue(research.Space in new_research)
31 self.assertTrue(research.ArtificialIntelligence not in new_research)