|
8
|
1 from unittest import TestCase
|
|
|
2
|
|
|
3 from gamelib.lab import Lab
|
|
14
|
4 from gamelib import research, products
|
|
8
|
5
|
|
|
6
|
|
|
7 class TestLab(TestCase):
|
|
14
|
8 def setUp(self):
|
|
|
9 self.lab = Lab()
|
|
|
10
|
|
8
|
11 def test_find_new_products(self):
|
|
14
|
12 new_products = self.lab.find_new_products(None)
|
|
|
13 self.assertTrue(products.MachineGun in new_products)
|
|
|
14 self.assertTrue(products.DoomsdayVirus not in new_products)
|
|
8
|
15
|
|
|
16 def test_find_new_research(self):
|
|
14
|
17 new_research = self.lab.find_new_research(None)
|
|
|
18 self.assertTrue(research.Tesla in new_research)
|
|
|
19 self.assertTrue(research.Space not in new_research)
|
