From 7059b4af6d651a819daf8b6ea838ae82f62287f0 Mon Sep 17 00:00:00 2001
From: Paul Sokolovsky <pfalcon@users.sourceforge.net>
Date: Sun, 26 Aug 2018 08:53:29 +0300
Subject: [PATCH] tests/uctypes_sizeof_od: Test for using OrderedDict as struct
descriptor
Just a copy of uctypes_sizeof.py with minimal changes.
---
tests/extmod/uctypes_sizeof_od.py | 48 +++++++++++++++++++++++++++
tests/extmod/uctypes_sizeof_od.py.exp | 7 ++++
2 files changed, 55 insertions(+)
create mode 100644 tests/extmod/uctypes_sizeof_od.py
create mode 100644 tests/extmod/uctypes_sizeof_od.py.exp
diff --git a/tests/extmod/uctypes_sizeof_od.py b/tests/extmod/uctypes_sizeof_od.py
new file mode 100644
index 000000000..192ee9152
--- /dev/null
+++ b/tests/extmod/uctypes_sizeof_od.py
@@ -0,0 +1,48 @@
+try:
+ from ucollections import OrderedDict
+ import uctypes
+except ImportError:
+ print("SKIP")
+ raise SystemExit
+
+desc = OrderedDict({
+ # arr is array at offset 0, of UINT8 elements, array size is 2
+ "arr": (uctypes.ARRAY | 0, uctypes.UINT8 | 2),
+ # arr2 is array at offset 0, size 2, of structures defined recursively
+ "arr2": (uctypes.ARRAY | 0, 2, {"b": uctypes.UINT8 | 0}),
+ "arr3": (uctypes.ARRAY | 2, uctypes.UINT16 | 2),
+ "arr4": (uctypes.ARRAY | 0, 2, {"b": uctypes.UINT8 | 0, "w": uctypes.UINT16 | 1}),
+ "sub": (0, {
+ 'b1': uctypes.BFUINT8 | 0 | 4 << uctypes.BF_POS | 4 << uctypes.BF_LEN,
+ 'b2': uctypes.BFUINT8 | 0 | 0 << uctypes.BF_POS | 4 << uctypes.BF_LEN,
+ }),
+})
+
+data = bytearray(b"01234567")
+
+S = uctypes.struct(uctypes.addressof(data), desc, uctypes.LITTLE_ENDIAN)
+
+print(uctypes.sizeof(S.arr))
+assert uctypes.sizeof(S.arr) == 2
+
+print(uctypes.sizeof(S.arr2))
+assert uctypes.sizeof(S.arr2) == 2
+
+print(uctypes.sizeof(S.arr3))
+
+try:
+ print(uctypes.sizeof(S.arr3[0]))
+except TypeError:
+ print("TypeError")
+
+print(uctypes.sizeof(S.arr4))
+assert uctypes.sizeof(S.arr4) == 6
+
+print(uctypes.sizeof(S.sub))
+assert uctypes.sizeof(S.sub) == 1
+
+# invalid descriptor
+try:
+ print(uctypes.sizeof([]))
+except TypeError:
+ print("TypeError")
diff --git a/tests/extmod/uctypes_sizeof_od.py.exp b/tests/extmod/uctypes_sizeof_od.py.exp
new file mode 100644
index 000000000..b35b11aa0
--- /dev/null
+++ b/tests/extmod/uctypes_sizeof_od.py.exp
@@ -0,0 +1,7 @@
+2
+2
+4
+TypeError
+6
+1
+TypeError
--
GitLab