Added test against multiple outputs.

Author: mvcisback

aiger/parser.py

@@ -9,7 +9,7 @@
 from bidict import bidict
 from toposort import toposort_flatten
 from uuid import uuid1
-from sortedcontainers import SortedSet, SortedDict
+from sortedcontainers import SortedList, SortedSet, SortedDict
 
 import aiger as A
 
@@ -60,7 +60,7 @@ class SymbolTable:
 class State:
     header: Optional[Header] = None
     inputs: SortedSet = attr.ib(factory=SortedSet)
-    outputs: SortedSet = attr.ib(factory=SortedSet)
+    outputs: SortedList = attr.ib(factory=SortedList)
     latches: SortedSet = attr.ib(factory=SortedSet)
     symbols: SymbolTable = attr.ib(factory=SymbolTable)
     comments: Optional[List[str]] = None
@@ -223,7 +223,7 @@ def parse_seq():
 
 def finish_table(table, keys):
     assert len(table) <= len(keys)
-    return bidict({table[i]: key for i, key in enumerate(keys)})
+    return {table[i]: key for i, key in enumerate(keys)}
 
 
 def parse(lines, to_aig: bool = True):

poetry.lock

@@ -77,6 +77,7 @@ def test_mutex_example_smoke():
 
 def test_degenerate_smoke():
     import aiger as A
+
     expr = A.BoolExpr(A.parse("""aag 0 0 0 1 0
 0
 """))
@@ -89,3 +90,9 @@ def test_degenerate_smoke():
 """)
     assert len(circ.node_map) == 0
     assert circ.inputs == circ.outputs == circ.latches == set()
+
+    circ = A.parse("""aag 0 0 0 2 0
+0
+0
+""")
+    assert not any(circ({})[0].values())