From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Thu, 17 Sep 2015 03:08:40 +0000 (+0800)
Subject: test/constant: use new API
X-Git-Tag: 24jan2021_ls180~2099^2~3^2~87
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=fd88b9b8a39311f3923470023c981ee7de87fab1;p=litex.git

test/constant: use new API
---

diff --git a/migen/test/test_constant.py b/migen/test/test_constant.py
index 11ac5f6b..8ad637fc 100644
--- a/migen/test/test_constant.py
+++ b/migen/test/test_constant.py
@@ -1,11 +1,11 @@
 import unittest
 
-from migen.fhdl.std import *
-from migen.test.support import SimCase, SimBench
+from migen import *
+from migen.test.support import SimCase
 
 
 class ConstantCase(SimCase, unittest.TestCase):
-    class TestBench(SimBench):
+    class TestBench(Module):
         def __init__(self):
             self.sigs = [
                 (Signal(3), Constant(0), 0),
@@ -18,11 +18,11 @@ class ConstantCase(SimCase, unittest.TestCase):
             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)
+        def gen():
+            for s, l, v in self.tb.sigs:
+                s = yield s
                 self.assertEqual(
                     s, int(v),
                     "got {}, want {} from literal {}".format(
                         s, v, l))
-        self.run_with(cb, 1)
+        self.run_with(gen())