move to common ALUHelpers for ShiftRot test_pipe_caller.py

diff --git a/src/soc/fu/shift_rot/test/test_pipe_caller.py b/src/soc/fu/shift_rot/test/test_pipe_caller.py
index db652c0873fea9ed47f44c4cea4eace7c467ac6b..98a719e9a7fb147661c6676b4995d0f49a190563 100644 (file)
--- a/src/soc/fu/shift_rot/test/test_pipe_caller.py
+++ b/src/soc/fu/shift_rot/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.shift_rot.pipeline import ShiftRotBasePipe
 from soc.fu.alu.alu_input_record import CompALUOpSubset
 from soc.fu.shift_rot.pipe_data import ShiftRotPipeSpec
@@ -60,30 +60,10 @@ 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'])
-    else:
-        yield alu.p.data_i.a.eq(0)
-    if 'rb' in inp:
-        yield alu.p.data_i.rb.eq(inp['rb'])
-    else:
-        yield alu.p.data_i.rb.eq(0)
-    if 'rc' in inp:
-        yield alu.p.data_i.rs.eq(inp['rc'])
-    else:
-        yield alu.p.data_i.rs.eq(0)
-
-    # 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.rb.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']))
-    else:
-        yield alu.p.data_i.xer_ca.eq(0)
+    yield from ALUHelpers.set_int_ra(alu, dec2, inp)
+    yield from ALUHelpers.set_int_rb(alu, dec2, inp)
+    yield from ALUHelpers.set_int_rc(alu, dec2, inp)
+    yield from ALUHelpers.set_xer_ca(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 5ffd130bbaf5ecee22ecba476e9b08a09ea5a370..b68e0739ecdc3d53aa31584941824024108560be 100644 (file)
--- a/src/soc/fu/test/common.py
+++ b/src/soc/fu/test/common.py
@@ -33,7 +33,13 @@ class ALUHelpers:
         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)
+            yield alu.p.data_i.rb.eq(data2)
+
+    def set_int_rc(alu, dec2, inp):
+        if 'rc' in inp:
+            yield alu.p.data_i.rc.eq(inp['rc'])
+        else:
+            yield alu.p.data_i.rc.eq(0)
 
     def set_xer_ca(alu, dec2, inp):
         if 'xer_ca' in inp: