X-Git-Url: https://git.libre-soc.org/?a=blobdiff_plain;f=src%2Fieee754%2Fpart%2Ftest%2Ftest_partsig.py;h=c9dfcb5216e608194ab4a835fc6ae407d2c32b1c;hb=73f2a4b0606065b9ff213bca1bd7c14acadf2a1d;hp=eb12b8a601cad98a12badbf151a0b45d68247a95;hpb=544ebd8d35df7df575e3ffd03e77c9ea4980f114;p=ieee754fpu.git

diff --git a/src/ieee754/part/test/test_partsig.py b/src/ieee754/part/test/test_partsig.py
index eb12b8a6..c9dfcb52 100644
--- a/src/ieee754/part/test/test_partsig.py
+++ b/src/ieee754/part/test/test_partsig.py
@@ -26,6 +26,7 @@ def create_simulator(module, traces, test_name):
 
 class TestAddMod(Elaboratable):
     def __init__(self, width, partpoints):
+        self.partpoints = partpoints
         self.a = PartitionedSignal(partpoints, width)
         self.b = PartitionedSignal(partpoints, width)
         self.add_output = Signal(width)
@@ -49,7 +50,8 @@ class TestAddMod(Elaboratable):
         m.d.comb += self.eq_output.eq(self.a == self.b)
         m.d.comb += self.ge_output.eq(self.a >= self.b)
         m.d.comb += self.add_output.eq(self.a + self.b)
-        m.d.comb += self.mux_out.eq(PMux(m, self.a, self.b, self.mux_sel))
+        ppts = self.partpoints
+        m.d.comb += self.mux_out.eq(PMux(m, ppts, self.mux_sel, self.a, self.b))
 
         return m