fix: array.array __getitem__ __setitem__ not defined

Signed-off-by: Jos Verlinde <jos_verlinde@hotmail.com>
Josverl · Nov 10, 2023 · 1741969 · 1741969
1 parent 5710d6c
commit 1741969
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/snippets/feat_stdlib/check_array.py b/snippets/feat_stdlib/check_array.py
@@ -0,0 +1,10 @@
+import array
+
+NUM_LEDS = 10
+
+ar = array.array("I", [0 for _ in range(NUM_LEDS)])
+
+x = ar[3] 
+# type: ignore # TODO: __getitem__ not defined 
+ar[3] = 0x12345678 
+# type: ignore # TODO: __setitem__ not defined
diff --git a/src/stubber/rst/lookup.py b/src/stubber/rst/lookup.py
@@ -515,6 +515,8 @@ class Fix:
     # ESPNow
     "ESPNow": "ESPNowBase,Iterator",
     "AIOESPNow": "ESPNow",
+    # array
+    "array": "List",
 }