From: whitequark <cz@m-labs.hk>
Date: Tue, 18 Dec 2018 18:02:21 +0000 (+0000)
Subject: back.pysim: implement *.
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=36341921c52c7d0e683b4186a04bf192ec17f9d7;p=nmigen.git

back.pysim: implement *.
---

diff --git a/nmigen/back/pysim.py b/nmigen/back/pysim.py
index d49ff41..206f199 100644
--- a/nmigen/back/pysim.py
+++ b/nmigen/back/pysim.py
@@ -113,6 +113,8 @@ class _RHSValueCompiler(AbstractValueTransformer):
                 return lambda state: normalize(lhs(state) +  rhs(state), shape)
             if value.op == "-":
                 return lambda state: normalize(lhs(state) -  rhs(state), shape)
+            if value.op == "*":
+                return lambda state: normalize(lhs(state) *  rhs(state), shape)
             if value.op == "&":
                 return lambda state: normalize(lhs(state) &  rhs(state), shape)
             if value.op == "|":
diff --git a/nmigen/test/test_sim.py b/nmigen/test/test_sim.py
index 3294b6c..f91c932 100644
--- a/nmigen/test/test_sim.py
+++ b/nmigen/test/test_sim.py
@@ -66,6 +66,12 @@ class SimulatorUnitTestCase(FHDLTestCase):
         self.assertStatement(stmt, [C(0,  4), C(1,  4)], C(-1,  4))
         self.assertStatement(stmt, [C(0,  4), C(10, 4)], C(-10, 5))
 
+    def test_mul(self):
+        stmt = lambda y, a, b: y.eq(a * b)
+        self.assertStatement(stmt, [C(2,  4), C(1,  4)], C(2,   8))
+        self.assertStatement(stmt, [C(2,  4), C(2,  4)], C(4,   8))
+        self.assertStatement(stmt, [C(7,  4), C(7,  4)], C(49,  8))
+
     def test_and(self):
         stmt = lambda y, a, b: y.eq(a & b)
         self.assertStatement(stmt, [C(0b1100, 4), C(0b1010, 4)], C(0b1000, 4))