diff --git a/tests/test_time.py b/tests/test_time.py index e35ef211b9c..0aef876ece0 100644 --- a/tests/test_time.py +++ b/tests/test_time.py @@ -5,7 +5,6 @@ # import unittest # from unittest import TestCase, mock - from mesa import Agent, Model from mesa.time import ( BaseScheduler, @@ -54,8 +53,6 @@ def step(self): self.model.log.append(self.unique_id) - - class MockModel(Model): def __init__( self, seed=None, shuffle=False, activation=STAGED, enable_kill_other_agent=False @@ -117,8 +114,8 @@ def __init__(self, some_number, seed=None, some_other_argument=5): self.some_number = some_number self.some_other_argument = some_other_argument -def test_some_model(): +def test_some_model(): some_model = SomeModel(5) some_model = SomeModel(5, seed=10) some_model = SomeModel(5, some_other_argument=10)