Add more uniquification tests

phanrahan · Feb 12, 2019 · a080f7c · a080f7c
1 parent 0cb0575
commit a080f7c
Show file tree

Hide file tree

Showing 3 changed files with 134 additions and 0 deletions.
diff --git a/tests/gold/uniquify_equal.json b/tests/gold/uniquify_equal.json
@@ -0,0 +1,36 @@
+{"top":"global.top",
+"namespaces":{
+  "global":{
+    "modules":{
+      "foo":{
+        "type":["Record",[
+          ["I","BitIn"],
+          ["O","Bit"]
+        ]],
+        "connections":[
+          ["self.O","self.I"]
+        ]
+      },
+      "top":{
+        "type":["Record",[
+          ["I","BitIn"],
+          ["O","Bit"]
+        ]],
+        "instances":{
+          "foo_inst0":{
+            "modref":"global.foo"
+          },
+          "foo_inst1":{
+            "modref":"global.foo"
+          }
+        },
+        "connections":[
+          ["self.I","foo_inst0.I"],
+          ["foo_inst1.I","foo_inst0.O"],
+          ["self.O","foo_inst1.O"]
+        ]
+      }
+    }
+  }
+}
+}
diff --git a/tests/gold/uniquify_unequal.json b/tests/gold/uniquify_unequal.json
@@ -0,0 +1,46 @@
+{"top":"global.top",
+"namespaces":{
+  "global":{
+    "modules":{
+      "foo":{
+        "type":["Record",[
+          ["I","BitIn"],
+          ["O","Bit"]
+        ]],
+        "connections":[
+          ["self.O","self.I"]
+        ]
+      },
+      "foo_unq1":{
+        "type":["Record",[
+          ["I",["Array",2,"BitIn"]],
+          ["O",["Array",2,"Bit"]]
+        ]],
+        "connections":[
+          ["self.O","self.I"]
+        ]
+      },
+      "top":{
+        "type":["Record",[
+          ["I","BitIn"],
+          ["O","Bit"]
+        ]],
+        "instances":{
+          "foo_inst0":{
+            "modref":"global.foo"
+          },
+          "foo_inst1":{
+            "modref":"global.foo_unq1"
+          }
+        },
+        "connections":[
+          ["self.I","foo_inst0.I"],
+          ["foo_inst1.I.0","foo_inst0.O"],
+          ["foo_inst1.I.1","foo_inst0.O"],
+          ["self.O","foo_inst1.O.0"]
+        ]
+      }
+    }
+  }
+}
+}
diff --git a/tests/test_uniquify.py b/tests/test_uniquify.py
@@ -1,4 +1,5 @@
 import magma as m
+from magma.testing import check_files_equal
 
 
 def test_verilog_field_uniquify():
@@ -11,3 +12,54 @@ def test_verilog_field_uniquify():
         assign C = A & B;\
     '''
     m.EndCircuit()
+
+def test_uniquify_equal():
+    foo = m.DefineCircuit("foo", "I", m.In(m.Bit), "O", m.Out(m.Bit))
+    m.wire(foo.I, foo.O)
+    m.EndCircuit()
+
+    bar = m.DefineCircuit("foo", "I", m.In(m.Bit), "O", m.Out(m.Bit))
+    m.wire(bar.I, bar.O)
+    m.EndCircuit()
+
+    top = m.DefineCircuit("top", "I", m.In(m.Bit), "O", m.Out(m.Bit))
+    curr = top.I
+    for circ in (foo, bar):
+        inst = circ()
+        m.wire(inst.I, curr)
+        curr = inst.O
+    m.wire(curr, top.O)
+    m.EndCircuit()
+
+    assert hash(foo) == hash(bar)
+
+    m.compile("build/uniquify_equal", top, output="coreir")
+    assert check_files_equal(__file__,
+                             "build/uniquify_equal.json",
+                             "gold/uniquify_equal.json")
+
+
+def test_uniquify_unequal():
+    foo = m.DefineCircuit("foo", "I", m.In(m.Bit), "O", m.Out(m.Bit))
+    m.wire(foo.I, foo.O)
+    m.EndCircuit()
+
+    bar = m.DefineCircuit("foo", "I", m.In(m.Bits(2)), "O", m.Out(m.Bits(2)))
+    m.wire(bar.I, bar.O)
+    m.EndCircuit()
+
+    top = m.DefineCircuit("top", "I", m.In(m.Bit), "O", m.Out(m.Bit))
+    foo_inst = foo()
+    m.wire(top.I, foo_inst.I)
+    bar_inst = bar()
+    m.wire(foo_inst.O, bar_inst.I[0])
+    m.wire(foo_inst.O, bar_inst.I[1])
+    m.wire(bar_inst.O[0], top.O)
+    m.EndCircuit()
+
+    assert hash(foo) != hash(bar)
+
+    m.compile("build/uniquify_unequal", top, output="coreir")
+    assert check_files_equal(__file__,
+                             "build/uniquify_unequal.json",
+                             "gold/uniquify_unequal.json")