def test_run(self): mdig.repository = ModelRepository() models = mdig.repository.get_models() fn = models['lifestage_test'] m = DispersalModel(fn) m._get_instance_w_smallest_reps_remaining = Mock() m._get_instance_w_smallest_reps_remaining.return_value = None m.run() self.assertEqual(m.active, False) # These are less than or equal because of VM clock being weird self.assertTrue(m.start_time <= datetime.datetime.now()) self.assertTrue(m.start_time <= m.end_time)