From d934db55a38831654f6f59a4d0eb71aa04c886ef Mon Sep 17 00:00:00 2001
From: Luke Kenneth Casson Leighton <lkcl@lkcl.net>
Date: Mon, 1 Jun 2020 12:38:45 +0100
Subject: [PATCH] add first version of ShiftRot CompUnit test

---
 .../compunits/test/test_shiftrot_compunit.py  | 102 ++++++++++++++++++
 1 file changed, 102 insertions(+)
 create mode 100644 src/soc/fu/compunits/test/test_shiftrot_compunit.py

diff --git a/src/soc/fu/compunits/test/test_shiftrot_compunit.py b/src/soc/fu/compunits/test/test_shiftrot_compunit.py
new file mode 100644
index 00000000..34935b5f
--- /dev/null
+++ b/src/soc/fu/compunits/test/test_shiftrot_compunit.py
@@ -0,0 +1,102 @@
+import unittest
+from soc.decoder.power_enums import (XER_bits, Function)
+
+# XXX bad practice: use of global variables
+from soc.fu.shift_rot.test.test_pipe_caller import ShiftRotTestCase
+from soc.fu.shift_rot.test.test_pipe_caller import test_data
+
+from soc.fu.compunits.compunits import ShiftRotFunctionUnit
+from soc.fu.compunits.test.test_compunit import TestRunner
+
+
+class ShiftRotTestRunner(TestRunner):
+    def __init__(self, test_data):
+        super().__init__(test_data, ShiftRotFunctionUnit, self,
+                         Function.SHIFT_ROT)
+
+    def get_cu_inputs(self, dec2, sim):
+        """naming (res) must conform to ShiftRotFunctionUnit input regspec
+        """
+        res = {}
+
+        # RA
+        reg1_ok = yield dec2.e.read_reg1.ok
+        if reg1_ok:
+            data1 = yield dec2.e.read_reg1.data
+            res['a'] = sim.gpr(data1).value
+
+        # RB (or immediate)
+        reg2_ok = yield dec2.e.read_reg2.ok
+        if reg2_ok:
+            data2 = yield dec2.e.read_reg2.data
+            res['rs'] = sim.gpr(data2).value
+
+        # RS (RC)
+        reg3_ok = yield dec2.e.read_reg3.ok
+        if reg3_ok:
+            data3 = yield dec2.e.read_reg3.data
+            res['rb'] = sim.gpr(data3).value
+
+        # XER.ca
+        carry = 1 if sim.spr['XER'][XER_bits['CA']] else 0
+        carry32 = 1 if sim.spr['XER'][XER_bits['CA32']] else 0
+        res['xer_ca'] = carry | (carry32<<1)
+
+        return res
+
+    def check_cu_outputs(self, res, dec2, sim, code):
+        """naming (res) must conform to ShiftRotFunctionUnit output regspec
+        """
+
+        # RT
+        out_reg_valid = yield dec2.e.write_reg.ok
+        if out_reg_valid:
+            write_reg_idx = yield dec2.e.write_reg.data
+            expected = sim.gpr(write_reg_idx).value
+            cu_out = res['o']
+            print(f"expected {expected:x}, actual: {cu_out:x}")
+            self.assertEqual(expected, cu_out, code)
+
+        rc = yield dec2.e.rc.data
+        op = yield dec2.e.insn_type
+        cridx_ok = yield dec2.e.write_cr.ok
+        cridx = yield dec2.e.write_cr.data
+
+        print ("check extra output", repr(code), cridx_ok, cridx)
+
+        if rc:
+            self.assertEqual(cridx_ok, 1, code)
+            self.assertEqual(cridx, 0, code)
+
+        # CR (CR0-7)
+        if cridx_ok:
+            cr_expected = sim.crl[cridx].get_range().value
+            cr_actual = res['cr0']
+            print ("CR", cridx, cr_expected, cr_actual)
+            self.assertEqual(cr_expected, cr_actual, "CR%d %s" % (cridx, code))
+
+        # XER.ca
+        cry_out = yield dec2.e.output_carry
+        if cry_out:
+            expected_carry = 1 if sim.spr['XER'][XER_bits['CA']] else 0
+            xer_ca = res['xer_ca']
+            real_carry = xer_ca & 0b1 # XXX CO not CO32
+            self.assertEqual(expected_carry, real_carry, code)
+            expected_carry32 = 1 if sim.spr['XER'][XER_bits['CA32']] else 0
+            real_carry32 = bool(xer_ca & 0b10) # XXX CO32
+            self.assertEqual(expected_carry32, real_carry32, code)
+
+        # TODO: XER.ov and XER.so
+        oe = yield dec2.e.oe.data
+        if oe:
+            xer_ov = res['xer_ov']
+            xer_so = res['xer_so']
+
+
+if __name__ == "__main__":
+    unittest.main(exit=False)
+    suite = unittest.TestSuite()
+    suite.addTest(ShiftRotTestRunner(test_data))
+
+    runner = unittest.TextTestRunner()
+    runner.run(suite)
-- 
2.30.2