Merge pull request #2098 from Shaikh-Ubaid/fix_cpython_cast

Fix unsigned integer cast for CPython

Author: certik

integration_tests/CMakeLists.txt

@@ -634,6 +634,7 @@ RUN(NAME test_str_comparison LABELS cpython llvm c)
 RUN(NAME test_bit_length     LABELS cpython llvm c)
 RUN(NAME str_to_list_cast    LABELS cpython llvm c)
 RUN(NAME cast_01             LABELS cpython llvm c)
+RUN(NAME cast_02             LABELS cpython llvm c)
 RUN(NAME test_sys_01         LABELS cpython llvm c NOFAST)
 RUN(NAME intent_01           LABELS cpython llvm)
 

integration_tests/cast_02.py

@@ -0,0 +1,42 @@
+from lpython import u8, u16, u32, u64
+
+def test_01():
+    x : u32 = u32(10)
+    print(x)
+    assert x == u32(10)
+
+    y: u16 = u16(x)
+    print(y)
+    assert y == u16(10)
+
+    z: u64 = u64(y)
+    print(z)
+    assert z == u64(10)
+
+    w: u8 = u8(z)
+    print(w)
+    assert w == u8(10)
+
+def test_02():
+    x : u64 = u64(11)
+    print(x)
+    assert x == u64(11)
+
+    y: u8 = u8(x)
+    print(y)
+    assert y == u8(11)
+
+    z: u16 = u16(y)
+    print(z)
+    assert z == u16(11)
+
+    w: u32 = u32(z)
+    print(w)
+    assert w == u32(11)
+
+
+def main0():
+    test_01()
+    test_02()
+
+main0()

src/runtime/lpython/lpython.py

@@ -17,14 +17,9 @@
 class UnsignedInteger:
     def __init__(self, bit_width, value):
         if isinstance(value, UnsignedInteger):
-            if bit_width != value.bit_width:
-                raise ValueError(f"Bit width mismatch: {bit_width} vs {value.bit_width}")
             value = value.value
-
-        if not (0 <= value < 2**bit_width):
-            raise ValueError(f"Value should be in range 0 to {2**bit_width-1} for a {bit_width}-bit unsigned integer.")
         self.bit_width = bit_width
-        self.value = value
+        self.value = value % (2**bit_width)
 
     def __bool__(self):
         return self.value != 0