def tearDown(self): Building.set_pm(self.old_pm)
def setUp(self): self.old_pm = Building._pm self.pm = MagicMock() Building.set_pm(self.pm)