diff --git a/tests/test_datacollector.py b/tests/test_datacollector.py index ff469fbfd03..5c2575df31b 100644 --- a/tests/test_datacollector.py +++ b/tests/test_datacollector.py @@ -132,6 +132,7 @@ class TestDataCollector(unittest.TestCase): def setUp(self): """Create the model and run it a set number of steps.""" self.model = MockModel() + self.model.datacollector.collect(self.model) for i in range(7): if i == 4: self.model.schedule.remove(self.model.schedule._agents[3])