From 2ec835f572cdee78c6c1ffd300bd159e0e35598a Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Tue, 13 Oct 2015 18:24:36 +0100
Subject: [PATCH] tests: Add more tests for viper 16/32-bit load/store, and
 ellipsis.

---
 tests/micropython/viper_misc.py            |  6 ++++++
 tests/micropython/viper_misc.py.exp        |  1 +
 tests/micropython/viper_ptr16_load.py      | 10 ++++++++++
 tests/micropython/viper_ptr16_load.py.exp  |  1 +
 tests/micropython/viper_ptr16_store.py     | 19 ++++++++++++++++++-
 tests/micropython/viper_ptr16_store.py.exp |  4 +++-
 tests/micropython/viper_ptr32_load.py      | 10 ++++++++++
 tests/micropython/viper_ptr32_load.py.exp  |  1 +
 tests/micropython/viper_ptr32_store.py     | 10 ++++++++++
 tests/micropython/viper_ptr32_store.py.exp |  1 +
 10 files changed, 61 insertions(+), 2 deletions(-)

diff --git a/tests/micropython/viper_misc.py b/tests/micropython/viper_misc.py
index e8b96b9c1..8bf6cc638 100644
--- a/tests/micropython/viper_misc.py
+++ b/tests/micropython/viper_misc.py
@@ -18,6 +18,12 @@ def viper_ret_none() -> int:
     return None
 print(viper_ret_none())
 
+# return Ellipsis as object
+@micropython.viper
+def viper_ret_ellipsis() -> object:
+    return ...
+print(viper_ret_ellipsis())
+
 # 3 args
 @micropython.viper
 def viper_3args(a:int, b:int, c:int) -> int:
diff --git a/tests/micropython/viper_misc.py.exp b/tests/micropython/viper_misc.py.exp
index 7859d39e5..4800050d1 100644
--- a/tests/micropython/viper_misc.py.exp
+++ b/tests/micropython/viper_misc.py.exp
@@ -1,6 +1,7 @@
 6
 3
 0
+Ellipsis
 6
 7
 20
diff --git a/tests/micropython/viper_ptr16_load.py b/tests/micropython/viper_ptr16_load.py
index 81a413c40..0b865eb9e 100644
--- a/tests/micropython/viper_ptr16_load.py
+++ b/tests/micropython/viper_ptr16_load.py
@@ -16,7 +16,17 @@ def memadd(src:ptr16, n:int) -> int:
         sum += src[i]
     return sum
 
+@micropython.viper
+def memadd2(src_in) -> int:
+    src = ptr16(src_in)
+    n = int(len(src_in)) >> 1
+    sum = 0
+    for i in range(n):
+        sum += src[i]
+    return sum
+
 b = bytearray(b'1234')
 print(b)
 print(get(b), get1(b))
 print(memadd(b, 2))
+print(memadd2(b))
diff --git a/tests/micropython/viper_ptr16_load.py.exp b/tests/micropython/viper_ptr16_load.py.exp
index caf475489..2d86b974c 100644
--- a/tests/micropython/viper_ptr16_load.py.exp
+++ b/tests/micropython/viper_ptr16_load.py.exp
@@ -1,3 +1,4 @@
 bytearray(b'1234')
 12849 13363
 26212
+26212
diff --git a/tests/micropython/viper_ptr16_store.py b/tests/micropython/viper_ptr16_store.py
index 94cde2bc6..5a5f25d17 100644
--- a/tests/micropython/viper_ptr16_store.py
+++ b/tests/micropython/viper_ptr16_store.py
@@ -4,16 +4,33 @@
 def set(dest:ptr16, val:int):
     dest[0] = val
 
+@micropython.viper
+def set1(dest:ptr16, val:int):
+    dest[1] = val
+
 @micropython.viper
 def memset(dest:ptr16, val:int, n:int):
     for i in range(n):
         dest[i] = val
 
+@micropython.viper
+def memset2(dest_in, val:int):
+    dest = ptr16(dest_in)
+    n = int(len(dest_in)) >> 1
+    for i in range(n):
+        dest[i] = val
+
 b = bytearray(4)
 print(b)
 
 set(b, 0x4242)
 print(b)
 
-memset(b, 0x4343, len(b) // 2)
+set1(b, 0x4343)
+print(b)
+
+memset(b, 0x4444, len(b) // 2)
+print(b)
+
+memset2(b, 0x4545)
 print(b)
diff --git a/tests/micropython/viper_ptr16_store.py.exp b/tests/micropython/viper_ptr16_store.py.exp
index 639a43f8f..31a6dfbe3 100644
--- a/tests/micropython/viper_ptr16_store.py.exp
+++ b/tests/micropython/viper_ptr16_store.py.exp
@@ -1,3 +1,5 @@
 bytearray(b'\x00\x00\x00\x00')
 bytearray(b'BB\x00\x00')
-bytearray(b'CCCC')
+bytearray(b'BBCC')
+bytearray(b'DDDD')
+bytearray(b'EEEE')
diff --git a/tests/micropython/viper_ptr32_load.py b/tests/micropython/viper_ptr32_load.py
index d552b9df6..4d8b3846d 100644
--- a/tests/micropython/viper_ptr32_load.py
+++ b/tests/micropython/viper_ptr32_load.py
@@ -15,7 +15,17 @@ def memadd(src:ptr32, n:int) -> int:
         sum += src[i]
     return sum
 
+@micropython.viper
+def memadd2(src_in) -> int:
+    src = ptr32(src_in)
+    n = int(len(src_in)) >> 2
+    sum = 0
+    for i in range(n):
+        sum += src[i]
+    return sum
+
 b = bytearray(b'\x12\x12\x12\x12\x34\x34\x34\x34')
 print(b)
 print(hex(get(b)), hex(get1(b)))
 print(hex(memadd(b, 2)))
+print(hex(memadd2(b)))
diff --git a/tests/micropython/viper_ptr32_load.py.exp b/tests/micropython/viper_ptr32_load.py.exp
index e7ce2d972..7ea37f1ca 100644
--- a/tests/micropython/viper_ptr32_load.py.exp
+++ b/tests/micropython/viper_ptr32_load.py.exp
@@ -1,3 +1,4 @@
 bytearray(b'\x12\x12\x12\x124444')
 0x12121212 0x34343434
 0x46464646
+0x46464646
diff --git a/tests/micropython/viper_ptr32_store.py b/tests/micropython/viper_ptr32_store.py
index b63bac9ee..973086e4a 100644
--- a/tests/micropython/viper_ptr32_store.py
+++ b/tests/micropython/viper_ptr32_store.py
@@ -13,6 +13,13 @@ def memset(dest:ptr32, val:int, n:int):
     for i in range(n):
         dest[i] = val
 
+@micropython.viper
+def memset2(dest_in, val:int):
+    dest = ptr32(dest_in)
+    n = int(len(dest_in)) >> 2
+    for i in range(n):
+        dest[i] = val
+
 b = bytearray(8)
 print(b)
 
@@ -24,3 +31,6 @@ print(b)
 
 memset(b, 0x44444444, len(b) // 4)
 print(b)
+
+memset2(b, 0x45454545)
+print(b)
diff --git a/tests/micropython/viper_ptr32_store.py.exp b/tests/micropython/viper_ptr32_store.py.exp
index 13b9f418f..de8d0ec57 100644
--- a/tests/micropython/viper_ptr32_store.py.exp
+++ b/tests/micropython/viper_ptr32_store.py.exp
@@ -2,3 +2,4 @@ bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00')
 bytearray(b'BBBB\x00\x00\x00\x00')
 bytearray(b'BBBBCCCC')
 bytearray(b'DDDDDDDD')
+bytearray(b'EEEEEEEE')
-- 
GitLab