Mercurial > sypikslang
diff gamelib/tests/test_lab.py @ 14:9d61abb3cfaf
Better subclass handling.
author | Jeremy Thurgood <firxen@gmail.com> |
---|---|
date | Sun, 06 May 2012 15:58:03 +0200 |
parents | 967480d228b6 |
children | 10d3db1f1e08 |
line wrap: on
line diff
--- a/gamelib/tests/test_lab.py Sun May 06 15:56:22 2012 +0200 +++ b/gamelib/tests/test_lab.py Sun May 06 15:58:03 2012 +0200 @@ -1,13 +1,19 @@ from unittest import TestCase from gamelib.lab import Lab +from gamelib import research, products class TestLab(TestCase): + def setUp(self): + self.lab = Lab() + def test_find_new_products(self): - lab = Lab() - print lab.find_new_products(None) + new_products = self.lab.find_new_products(None) + self.assertTrue(products.MachineGun in new_products) + self.assertTrue(products.DoomsdayVirus not in new_products) def test_find_new_research(self): - lab = Lab() - print lab.find_new_research(None) + new_research = self.lab.find_new_research(None) + self.assertTrue(research.Tesla in new_research) + self.assertTrue(research.Space not in new_research)