From: Robert Jordens <jordens@gmail.com>
Date: Sun, 6 Sep 2015 23:51:59 +0000 (-0600)
Subject: add unittests for Constant
X-Git-Tag: 24jan2021_ls180~2099^2~3^2~88
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=74c9159a0148f88375e461fe6d8bd707c7e2f669;p=litex.git

add unittests for Constant
---

diff --git a/migen/test/test_constant.py b/migen/test/test_constant.py
new file mode 100644
index 00000000..11ac5f6b
--- /dev/null
+++ b/migen/test/test_constant.py
@@ -0,0 +1,28 @@
+import unittest
+
+from migen.fhdl.std import *
+from migen.test.support import SimCase, SimBench
+
+
+class ConstantCase(SimCase, unittest.TestCase):
+    class TestBench(SimBench):
+        def __init__(self):
+            self.sigs = [
+                (Signal(3), Constant(0), 0),
+                (Signal(3), Constant(5), 5),
+                (Signal(3), Constant(1, 2), 1),
+                (Signal(3), Constant(-1, 7), 7),
+                (Signal(3), Constant(0b10101)[:3], 0b101),
+                (Signal(3), Constant(0b10101)[1:4], 0b10),
+            ]
+            self.comb += [a.eq(b) for a, b, c in self.sigs]
+
+    def test_comparisons(self):
+        def cb(tb, tbp):
+            for s, l, v in tb.sigs:
+                s = tbp.simulator.rd(s)
+                self.assertEqual(
+                    s, int(v),
+                    "got {}, want {} from literal {}".format(
+                        s, v, l))
+        self.run_with(cb, 1)