Fix bigint support (#26)

regressed since 65b3c1f
aheui · Mar 31, 2024 · 9e09698 · 9e09698
1 parent 0fb41f3
commit 9e09698
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/aheui/aheui.py b/aheui/aheui.py
@@ -104,7 +104,7 @@ def __init__(self):
         self.size = 0
 
     def push(self, value):
-        # assert(isinstance(value, int))
+        # assert(isinstance(value, bigint.Int))
         node = Link(self.head, value)
         self.head = node
         self.size += 1
@@ -129,6 +129,7 @@ def __init__(self):
         self.size = 0
 
     def push(self, value):
+        # assert(isinstance(value, bigint.Int))
         tail = self.tail
         tail.value = value
         new = Link(None)
@@ -154,10 +155,10 @@ class Port(LinkedList):
     def __init__(self):
         self.head = None
         self.size = 0
-        self.last_push = 0
+        self.last_push = bigint.fromint(0)
 
     def push(self, value):
-        # assert(isinstance(value, int))
+        # assert(isinstance(value, bigint.Int))
         node = Link(self.head, value)
         self.head = node
         self.size += 1

diff --git a/aheui/int/bigint.py b/aheui/int/bigint.py
@@ -3,6 +3,9 @@
 from rpython.rlib.rbigint import rbigint
 
 
+Int = rbigint
+
+
 def fromstr(s):
     return rbigint.fromstr(s)
 

diff --git a/aheui/int/smallint.py b/aheui/int/smallint.py
@@ -1,3 +1,5 @@
+Int = int
+
 
 def fromstr(s):
     return int(s)