Merge pull request #1860 from godlygeek/int_to_bytes_overflow_checks

Add overflow checks for int to bytes conversions

1 files changed, 35 insertions, 0 deletions

diff --git a/tests/basics/array_overflow.py b/tests/basics/array_overflow.py
new file mode 100644
index 000000000..ba3eb7f06
--- /dev/null
+++ b/tests/basics/array_overflow.py
@@ -0,0 +1,35 @@
+import skip_if
+skip_if.no_bigint()
+
+try:
+    from array import array
+except ImportError:
+    print("SKIP")
+    raise SystemExit
+
+def test_array_overflow(typecode, val):
+    try:
+        print(array(typecode, [val]))
+    except OverflowError:
+        print('OverflowError')
+
+# small int -1
+test_array_overflow('Q', -1)
+test_array_overflow('L', -1)
+test_array_overflow('I', -1)
+test_array_overflow('H', -1)
+test_array_overflow('B', -1)
+
+# big int -1
+test_array_overflow('Q', -2**64 // 2**64)
+test_array_overflow('L', -2**64 // 2**64)
+test_array_overflow('I', -2**64 // 2**64)
+test_array_overflow('H', -2**64 // 2**64)
+test_array_overflow('B', -2**64 // 2**64)
+
+# big int 2**63
+test_array_overflow('q', 2**63)
+test_array_overflow('l', 2**63)
+test_array_overflow('i', 2**63)
+test_array_overflow('h', 2**63)
+test_array_overflow('b', 2**63)