Skip to content

Commit

Permalink
Update test_time.py
Browse files Browse the repository at this point in the history
  • Loading branch information
quaquel committed Sep 13, 2024
1 parent cf24f15 commit 7af339d
Showing 1 changed file with 17 additions and 1 deletion.
18 changes: 17 additions & 1 deletion tests/test_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,8 +54,23 @@ def step(self):


class SomeModel(Model):
def __init__(self, seed=None):
def __init__(self, seed=None, shuffle=False, activation=STAGED, enable_kill_other_agent=False):
super().__init__(seed=seed)
self.log = []
self.enable_kill_other_agent = enable_kill_other_agent
self.shuffle = shuffle

match activation:
case "random":
pass
case "staged":
pass
case "simultaneous":
pass
case "random_by_type":
pass
case _:
raise ValueError(f"Unknown activation type: {activation}")


class MockModel(Model):
Expand Down Expand Up @@ -119,6 +134,7 @@ def test_no_shuffle(self):
Testing the staged activation without shuffling.
"""
a = SomeModel()
print(a)
model = MockModel(shuffle=False)
model.step()
model.step()
Expand Down

0 comments on commit 7af339d

Please sign in to comment.