Fix tests

projectmesa · Sep 26, 2024 · f3c5696 · f3c5696
1 parent 8244dac
commit f3c5696
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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])