Adding error message when PA type is invalid.

Author: Dorian Birraux

wolframclient/language/array.py

@@ -16,9 +16,11 @@ def __init__(self, array, type, shape=None):
         self.array = array
         self.shape = shape or (len(array),)
         self.type = type
-        try:
-            self.struct = constants.STRUCT_MAPPING[type]
-        except KeyError:
+        self._valid_type_or_fail(type)
+        self.struct = constants.STRUCT_MAPPING[type]
+
+    def _valid_type_or_fail(self, type):
+        if type not in constants.STRUCT_MAPPING:
             raise WolframLanguageException(
                 "Type %s is not one of the supported array types: %s."
                 % (type, ", ".join(constants.STRUCT_MAPPING.keys()))
@@ -35,4 +37,10 @@ def __len__(self):
 
 
 class PackedArray(NumericArray):
-    pass
+
+    def _valid_type_or_fail(self, type):
+        if type not in constants.VALID_PACKED_ARRAY_LABEL_TYPES:
+            raise WolframLanguageException(
+                "Type %s is not one of the supported packed array types: %s."
+                % (type, ', '.join(constants.VALID_PACKED_ARRAY_LABEL_TYPES_TUPLE))
+            )

wolframclient/serializers/wxfencoder/constants.py

@@ -91,6 +91,9 @@ def _bytes(value):
     )
 )
 
+VALID_PACKED_ARRAY_LABEL_TYPES_TUPLE = ('Integer8', 'Integer16', 'Integer32', 'Integer64', 'Real32', 'Real64', 'ComplexReal32', 'ComplexReal64')
+VALID_PACKED_ARRAY_LABEL_TYPES = frozenset(VALID_PACKED_ARRAY_LABEL_TYPES_TUPLE)
+
 STRUCT_MAPPING = Settings(
     Integer8=struct.Struct(b"<b"),
     UnsignedInteger8=struct.Struct(b"<B"),
@@ -104,4 +107,4 @@ def _bytes(value):
     Real64=struct.Struct(b"<d"),
     ComplexReal32=struct.Struct(b"<f"),
     ComplexReal64=struct.Struct(b"<d"),
-)
+)