import pickle from kyupy.circuit import Circuit, Node, Line from kyupy import verilog, bench from kyupy.techlib import SAED32 def test_lines(): c = Circuit() n1 = Node(c, 'n1') n2 = Node(c, 'n2') line = Line(c, n1, n2) assert line.driver == n1 assert line.reader == n2 assert line.driver_pin == 0 assert line.reader_pin == 0 assert n1.outs[0] == line assert n2.ins[0] == line line2 = Line(c, n1, (n2, 2)) assert line2.driver == n1 assert line2.reader == n2 assert line2.driver_pin == 1 assert line2.reader_pin == 2 assert n1.outs[0] == line assert n1.outs[1] == line2 assert n2.ins[1] is None assert n2.ins[2] == line2 line3 = Line(c, n1, n2) assert line3.driver_pin == 2 assert line3.reader_pin == 1 assert n1.outs[2] == line3 assert n2.ins[1] == line3 assert n2.ins[2] == line2 assert len(c.lines) == 3 line3.remove() assert len(c.lines) == 2 assert c.lines[0].index == 0 assert c.lines[1].index == 1 assert n1.outs[2] is None assert n2.ins[1] is None assert n2.ins[2] == line2 def test_circuit(): c = Circuit() in1 = Node(c, 'in1', 'buf') in2 = Node(c, 'in2', 'buf') out1 = Node(c, 'out1', 'buf') assert 'in1' in c.cells assert 'and1' not in c.cells c.io_nodes[0] = in1 c.io_nodes[1] = in2 c.io_nodes[2] = out1 and1 = Node(c, 'and1', kind='and') Line(c, in1, and1) Line(c, in2, and1) Line(c, and1, out1) assert len(in1.ins) == 0 assert len(in1.outs) == 1 assert len(in2.outs) == 1 assert in1.outs[0].reader == and1 assert in1.outs[0].driver == in1 assert len(and1.ins) == 2 assert len(and1.outs) == 1 or1 = Node(c, 'or1', 'or') Line(c, and1, (or1, 1)) or2 = Node(c, 'or2', 'or') or3 = Node(c, 'or3', 'or') assert or2.index == 5 assert or3.index == 6 assert len(c.nodes) == 7 or2.remove() or3 = c.cells['or3'] assert or3.index == 5 assert 'or2' not in c.cells assert len(c.nodes) == 6 c.cells['or3'].remove() assert 'or3' not in c.cells assert len(c.nodes) == 5 repr(c) str(c) for n in c.topological_order(): repr(n) def test_pickle(mydir): c = verilog.load(mydir / 'b15_4ig.v.gz', tlib=SAED32) assert c is not None cs = pickle.dumps(c) assert cs is not None c2 = pickle.loads(cs) assert c == c2 def test_substitute(): c = bench.parse('input(i1, i2, i3, i4, i5) output(o1) aoi=AOI221(i1, i2, i3, i4, i5) o1=not(aoi)') assert len(c.cells) == 2 assert len(c.io_nodes) == 6 aoi221_impl = bench.parse('input(in1, in2, in3, in4, in5) output(q) a1=and(in1, in2) a2=and(in3, in4) q=or(a1, a2, in5)') assert len(aoi221_impl.cells) == 3 assert len(aoi221_impl.io_nodes) == 6 c.substitute(c.cells['aoi'], aoi221_impl) assert len(c.cells) == 4 assert len(c.io_nodes) == 6 def test_resolve(mydir): c = verilog.load(mydir / 'b15_4ig.v.gz', tlib=SAED32) s_names = [n.name for n in c.s_nodes] c.resolve_tlib_cells(SAED32) s_names_prim = [n.name for n in c.s_nodes] assert s_names == s_names_prim, 'resolve_tlib_cells does not preserve names or order of s_nodes'