class ActorTest(unittest.TestCase):
##
# Generate World and disease for testing purposes
def setUp(self):
self.world_one = World(10, 15)
self.disease_one = Disease()
self.world_one.addObject(self.disease_one, 5, 10)
def test_constructor(self):
self.assertEqual(self.disease_one.getGrowthCondition(), (0, 0, 0))
def test_getStrenght(self):
self.assertEqual(self.disease_one.getStrength(), 1)
def test_getQuadrant(self):
self.assertEqual(self.disease_one.getQuadrant(), 3)
def test_setStrength(self):
self.disease_one.setStrength(2)
self.assertEqual(self.disease_one.getStrength(), 2)
def mousePressed(self, e):
x, y = self.wvmap.viewportToWindow(e.x, e.y)
disease = Disease()
disease.addedToWorld(self.world)
disease.setLocation(x, y)
quad = disease.getQuadrant()
self.diseases.append(disease)
if quad >= 0 or quad <= 3:
disease.setGrowthCondition(
self.world.getTemp(quad) - 1,
self.world.getTemp(quad) + 1, 2.0)
self.dict[(round(disease.getX()),
round(disease.getY()))] = disease.getStrength()
