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()