def testSetNextVolume(self): explosion = Explosion(2) self.assertEquals(ExplosionVolume_Normal, explosion.getVolume()) explosion.setNextVolume() self.assertEquals(ExplosionVolume_Hight, explosion.getVolume()) explosion.setNextVolume() self.assertEquals(ExplosionVolume_Low, explosion.getVolume()) explosion.setNextVolume() self.assertEquals(ExplosionVolume_Normal, explosion.getVolume())
def testSetNextVolumeOnBlowingExplosion(self): explosion = Explosion(2) explosion.setBlowingObject(StubBlowingObject()) explosion.blow() self.assertRaises(ExplosionError, lambda: explosion.setNextVolume())