Update python simulator to return BitVector instead of bool list

phanrahan · Aug 2, 2018 · 9942fab · 9942fab
1 parent 3c2a31b
commit 9942fab
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/magma/simulator/python_simulator.py b/magma/simulator/python_simulator.py
@@ -109,15 +109,15 @@ def value_initialized(self, bit):
 
     def get_value(self, bit):
         if isinstance(bit, ArrayType):
-            return [self.get_value(b) for b in bit]
+            return BitVector([self.get_value(b) for b in bit])
 
         if bit.isinput():
             bit = bit.value()
 
         if bit.const():
-            return True if bit is VCC else False
+            return BitVector(True, 1) if bit is VCC else BitVector(False, 1)
 
-        return self.value_map[bit]
+        return BitVector(self.value_map[bit], 1)
 
     def set_value(self, bit, newval):
         if isinstance(bit, ArrayType):