More tests to make codecov green

projectmesa · Aug 29, 2024 · bca99d0 · bca99d0
1 parent 2958838
commit bca99d0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -9,7 +9,9 @@ def test_model_set_up():
     assert model.current_id == 0
     assert model.current_id + 1 == model.next_id()
     assert model.current_id == 1
+    assert model.steps == 0
     model.step()
+    assert model.steps == 1
 
 
 def test_running():
@@ -18,12 +20,12 @@ class TestModel(Model):
 
         def step(self):
             """Increase steps until 10."""
-            self.steps += 1
             if self.steps == 10:
                 self.running = False
 
     model = TestModel()
     model.run_model()
+    assert model.steps == 10
 
 
 def test_seed(seed=23):