convert CR test to accumulator style

[soc.git] / src / soc / fu / test / common.py

diff --git a/src/soc/fu/test/common.py b/src/soc/fu/test/common.py
index b4bceb1b43bd7adcfc66ebd45f92f7ccaa4a8f03..99dd1d5c13b38c34b23c0ffa921257c5a388b299 100644 (file)
--- a/src/soc/fu/test/common.py
+++ b/src/soc/fu/test/common.py
@@ -19,9 +19,11 @@ class TestAccumulatorBase:
             if n.startswith("case_") and callable(v):
                 v(self)
 
-    def add_case(self, prog, initial_regs=None, initial_sprs=None):
+    def add_case(self, prog, initial_regs=None, initial_sprs=None,
+                        initial_cr=0):
         test_name = inspect.stack()[1][3] # name of caller of this function
-        tc = TestCase(prog, test_name, initial_regs, initial_sprs)
+        tc = TestCase(prog, test_name, 
+                      regs=initial_regs, sprs=initial_sprs, cr=initial_cr)
         self.test_data.append(tc)