Add exercises for m5 dig

1 files changed, 6 insertions, 1 deletions

diff --git a/sem6/dig/generate_test_file.py b/sem6/dig/generate_test_file.py
index 5a951f2..b97cbe6 100755
--- a/sem6/dig/generate_test_file.py
+++ b/sem6/dig/generate_test_file.py
@@ -113,7 +113,12 @@ architecture behavior of test_{name} is
     def wait():
         tof(f"wait for {td.get('teststep', 1)} fs;")
 
-    for tindex, t in enumerate(td["testin"]):
+    if "testin" in td:
+        enum = enumerate(td["testin"])
+    else:
+        enum = enumerate([1] * td["testcount"])
+
+    for tindex, t in enum:
         # Add clock
         if "clk" in td:
             tof(f"in_{td['clk']} <= '0';")