convert to individual PipeSpecs for each pipeline

[soc.git] / src / soc / fu / branch / test / test_pipe_caller.py

diff --git a/src/soc/fu/branch/test/test_pipe_caller.py b/src/soc/fu/branch/test/test_pipe_caller.py
index 53758b1df6839256da681e026d5bdda938fc30b6..75cac8090f561b80d86defbccf3fc05c77353755 100644 (file)
--- a/src/soc/fu/branch/test/test_pipe_caller.py
+++ b/src/soc/fu/branch/test/test_pipe_caller.py
@@ -13,7 +13,6 @@ from soc.decoder.isa.all import ISA
 
 
 from soc.fu.branch.pipeline import BranchBasePipe
-from soc.fu.branch.br_input_record import CompBROpSubset
 from soc.fu.branch.pipe_data import BranchPipeSpec
 import random
 
@@ -99,7 +98,7 @@ class BranchTestCase(FHDLTestCase):
                                  initial_cr=cr)
 
     def test_ilang(self):
-        rec = CompBROpSubset()
+        rec = BranchPipeSpec.opsubsetkls()
 
         pspec = BranchPipeSpec(id_wid=2, op_wid=get_rec_width(rec))
         alu = BranchBasePipe(pspec)
@@ -122,7 +121,7 @@ class TestRunner(FHDLTestCase):
 
         m.submodules.pdecode2 = pdecode2 = PowerDecode2(pdecode)
 
-        rec = CompBROpSubset()
+        rec = BranchPipeSpec.opsubsetkls()
 
         pspec = BranchPipeSpec(id_wid=2, op_wid=get_rec_width(rec))
         m.submodules.branch = branch = BranchBasePipe(pspec)