code-munge test_pipe_caller for ALU,

plan to remove duplicated code

diff --git a/src/soc/fu/alu/test/test_pipe_caller.py b/src/soc/fu/alu/test/test_pipe_caller.py
index 829a304fed0e74fa19b4199dbe3d183efc3a9c57..400afc980dd37c2581d57564ca8cc1344cfcd9bf 100644 (file)
--- a/src/soc/fu/alu/test/test_pipe_caller.py
+++ b/src/soc/fu/alu/test/test_pipe_caller.py
@@ -12,7 +12,7 @@ from soc.simulator.program import Program
 from soc.decoder.isa.all import ISA
 
 
-from soc.fu.test.common import TestCase
+from soc.fu.test.common import (TestCase, ALUHelpers)
 from soc.fu.alu.pipeline import ALUBasePipe
 from soc.fu.alu.pipe_data import ALUPipeSpec
 import random
@@ -60,24 +60,11 @@ def set_alu_inputs(alu, dec2, sim):
     # and place it into data_i.b
 
     inp = yield from get_cu_inputs(dec2, sim)
-    if 'ra' in inp:
-        yield alu.p.data_i.a.eq(inp['ra'])
-    if 'rb' in inp:
-        yield alu.p.data_i.b.eq(inp['rb'])
-
-    # If there's an immediate, set the B operand to that
-    imm_ok = yield dec2.e.imm_data.imm_ok
-    if imm_ok:
-        data2 = yield dec2.e.imm_data.imm
-        yield alu.p.data_i.b.eq(data2)
-
-    if 'xer_ca' in inp:
-        yield alu.p.data_i.xer_ca.eq(inp['xer_ca'])
-        print ("extra inputs: CA/32", bin(inp['xer_ca']))
-    if 'xer_so' in inp:
-        so = inp['xer_so']
-        print ("extra inputs: so", so)
-        yield alu.p.data_i.xer_so.eq(so)
+    yield from ALUHelpers.set_int_ra(alu, dec2, inp)
+    yield from ALUHelpers.set_int_rb(alu, dec2, inp)
+
+    yield from ALUHelpers.set_xer_ca(alu, dec2, inp)
+    yield from ALUHelpers.set_xer_so(alu, dec2, inp)
 
 
 # This test bench is a bit different than is usual. Initially when I

diff --git a/src/soc/fu/test/common.py b/src/soc/fu/test/common.py
index dc3cd9c3537dca5ebcf6ffa9cb35ff1f894b4b2c..9d034f4b839198c618f52edce79ad57923109c9e 100644 (file)
--- a/src/soc/fu/test/common.py
+++ b/src/soc/fu/test/common.py
@@ -16,3 +16,30 @@ class TestCase:
         self.cr = cr
         self.mem = mem
         self.msr = msr
+
+class ALUHelpers:
+
+    def set_int_ra(alu, dec2, inp):
+        if 'ra' in inp:
+            yield alu.p.data_i.ra.eq(inp['ra'])
+
+    def set_int_rb(alu, dec2, inp):
+        if 'rb' in inp:
+            yield alu.p.data_i.rb.eq(inp['rb'])
+        # If there's an immediate, set the B operand to that
+        imm_ok = yield dec2.e.imm_data.imm_ok
+        if imm_ok:
+            data2 = yield dec2.e.imm_data.imm
+            yield alu.p.data_i.b.eq(data2)
+
+    def set_xer_ca(alu, dec2, inp):
+        if 'xer_ca' in inp:
+            yield alu.p.data_i.xer_ca.eq(inp['xer_ca'])
+            print ("extra inputs: CA/32", bin(inp['xer_ca']))
+
+    def set_xer_so(alu, dec2, inp):
+        if 'xer_so' in inp:
+            so = inp['xer_so']
+            print ("extra inputs: so", so)
+            yield alu.p.data_i.xer_so.eq(so)
+