From fd88b9b8a39311f3923470023c981ee7de87fab1 Mon Sep 17 00:00:00 2001 From: Sebastien Bourdeauducq Date: Thu, 17 Sep 2015 11:08:40 +0800 Subject: [PATCH] test/constant: use new API --- migen/test/test_constant.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/migen/test/test_constant.py b/migen/test/test_constant.py index 11ac5f6bc..8ad637fc7 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())