From 7af339de5e0cde0dfe86d99fb793014cffb2420e Mon Sep 17 00:00:00 2001
From: Jan Kwakkel <j.h.kwakkel@tudelft.nl>
Date: Fri, 13 Sep 2024 21:16:14 +0200
Subject: [PATCH] Update test_time.py

---
 tests/test_time.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/tests/test_time.py b/tests/test_time.py
index 5d0d4a7413a..45d00741982 100644
--- a/tests/test_time.py
+++ b/tests/test_time.py
@@ -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):
@@ -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()