def test_null_bitmask(self):
mdig.repository = ModelRepository()
models = mdig.repository.get_models()
fn = models['lifestage_test']
m = DispersalModel(fn)
for i in m.get_instances():
i.null_bitmask = Mock()
m.null_bitmask()
for i in m.get_instances():
self.assertEqual(i.null_bitmask.call_count, 1)
self.assertEqual(i.null_bitmask.call_args[0][0], True)
# check parameter is passed
m.null_bitmask(generate=False)
for i in m.get_instances():
self.assertEqual(i.null_bitmask.call_count, 2)
self.assertEqual(i.null_bitmask.call_args[0][0], False)
# check disabled instances are not touched
for i in m.get_instances():
i.enabled = False
m.null_bitmask()
for i in m.get_instances():
self.assertEqual(i.null_bitmask.call_count, 2)
def test_null_bitmask(self):
mdig.repository = ModelRepository()
models = mdig.repository.get_models()
fn = models['lifestage_test']
m = DispersalModel(fn)
for i in m.get_instances():
i.null_bitmask = Mock()
m.null_bitmask()
for i in m.get_instances():
self.assertEqual(i.null_bitmask.call_count,1)
self.assertEqual(i.null_bitmask.call_args[0][0], True)
# check parameter is passed
m.null_bitmask(generate=False)
for i in m.get_instances():
self.assertEqual(i.null_bitmask.call_count,2)
self.assertEqual(i.null_bitmask.call_args[0][0], False)
# check disabled instances are not touched
for i in m.get_instances():
i.enabled=False
m.null_bitmask()
for i in m.get_instances():
self.assertEqual(i.null_bitmask.call_count,2)
