genlib/coding, test/test_coding: unittests

diff --git a/migen/genlib/coding.py b/migen/genlib/coding.py
index ed60e2f3c1d463d8962bafa8a92bc2ca1676396e..94a224ea25707f243fe234beafb781b0809f27b3 100644 (file)
--- a/migen/genlib/coding.py
+++ b/migen/genlib/coding.py
@@ -37,15 +37,3 @@ class Decoder(Module):
 
 class PriorityDecoder(Decoder):
        pass # same
-
-def _main():
-       from migen.fhdl import verilog
-       e = Encoder(8)
-       print(verilog.convert(e, ios={e.i, e.o, e.n}))
-       pe = PriorityEncoder(8)
-       print(verilog.convert(pe, ios={pe.i, pe.o, pe.n}))
-       d = Decoder(8)
-       print(verilog.convert(d, ios={d.i, d.n, d.o}))
-
-if __name__ == "__main__":
-       _main()

diff --git a/migen/test/test_coding.py b/migen/test/test_coding.py
new file mode 100644 (file)
index 0000000..de09805
--- /dev/null
+++ b/migen/test/test_coding.py
@@ -0,0 +1,82 @@
+import unittest
+
+from migen.fhdl.std import *
+from migen.genlib.coding import *
+
+from migen.test.support import SimCase, SimBench
+
+class EncCase(SimCase):
+       class TestBench(SimBench):
+               def __init__(self):
+                       self.submodules.dut = Encoder(8)
+
+       def test_sizes(self):
+               self.assertEqual(flen(self.tb.dut.i), 8)
+               self.assertEqual(flen(self.tb.dut.o), 3)
+               self.assertEqual(flen(self.tb.dut.n), 1)
+
+       def test_run_sequence(self):
+               seq = list(range(1<<8))
+               cur = None
+               def cb(tb, s):
+                       if seq:
+                               s.wr(tb.dut.i, seq.pop(0))
+                       i = s.rd(tb.dut.i)
+                       if s.rd(tb.dut.n):
+                               self.assertNotIn(i, [1<<i for i in range(8)])
+                       else:
+                               o = s.rd(tb.dut.o)
+                               self.assertEqual(i, 1<<o)
+               self.run_with(cb, 256)
+
+class PrioEncCase(SimCase):
+       class TestBench(SimBench):
+               def __init__(self):
+                       self.submodules.dut = PriorityEncoder(8)
+
+       def test_sizes(self):
+               self.assertEqual(flen(self.tb.dut.i), 8)
+               self.assertEqual(flen(self.tb.dut.o), 3)
+               self.assertEqual(flen(self.tb.dut.n), 1)
+
+       def test_run_sequence(self):
+               seq = list(range(1<<8))
+               cur = None
+               def cb(tb, s):
+                       if seq:
+                               s.wr(tb.dut.i, seq.pop(0))
+                       i = s.rd(tb.dut.i)
+                       if s.rd(tb.dut.n):
+                               self.assertEqual(i, 0)
+                       else:
+                               o = s.rd(tb.dut.o)
+                               if o > 0:
+                                       self.assertEqual(i & 1<<(o - 1), 0)
+                               self.assertGreaterEqual(i, 1<<o)
+               self.run_with(cb, 256)
+
+class DecCase(SimCase):
+       class TestBench(SimBench):
+               def __init__(self):
+                       self.submodules.dut = Decoder(8)
+
+       def test_sizes(self):
+               self.assertEqual(flen(self.tb.dut.i), 3)
+               self.assertEqual(flen(self.tb.dut.o), 8)
+               self.assertEqual(flen(self.tb.dut.n), 1)
+
+       def test_run_sequence(self):
+               seq = list(range(8*2))
+               cur = None
+               def cb(tb, s):
+                       if seq:
+                               i = seq.pop()
+                               s.wr(tb.dut.i, i//2)
+                               s.wr(tb.dut.n, i%2)
+                       i = s.rd(tb.dut.i)
+                       o = s.rd(tb.dut.o)
+                       if s.rd(tb.dut.n):
+                               self.assertEqual(o, 0)
+                       else:
+                               self.assertEqual(o, 1<<i)
+               self.run_with(cb, 256)