[CO] Possibility to have "None" in dict of pyleecan type (for HoleUD.magnet_dict)

Author: BonneelP

pyleecan/Classes/_check.py

@@ -167,31 +167,32 @@ def check_type(var_name, value, expect_type, type_value):
             raise CheckTypeError(
                 "For " + var_name + " : Dict expected, " + type_value + " given"
             )
-        # Check type of value element
+        # Check type of value element (None or expected type)
         expect_type = expect_type[1:-1]
         for key, element in value.items():
-            type_value = type(element).__name__
-            # Check if it's the expected type
-            if not type_value == expect_type:
-                # Check if object inherit from the expected type
-                mother = element.__class__.__bases__  # List of inherited class
-                find = False  # Store the result of the research
-                while mother != ():  # Every class inherit from FrozenClass
-                    if mother[0].__name__ == expect_type:
-                        find = True
-                        break
-                    else:
-                        mother = mother[0].__bases__
-                if not find:  # Not the good type
-                    raise CheckTypeError(
-                        "For "
-                        + var_name
-                        + " : "
-                        + expect_type
-                        + " expected, "
-                        + type_value
-                        + " given"
-                    )
+            if element is not None:
+                type_value = type(element).__name__
+                # Check if it's the expected type
+                if not type_value == expect_type:
+                    # Check if object inherit from the expected type
+                    mother = element.__class__.__bases__  # List of inherited class
+                    find = False  # Store the result of the research
+                    while mother != ():  # Every class inherit from FrozenClass
+                        if mother[0].__name__ == expect_type:
+                            find = True
+                            break
+                        else:
+                            mother = mother[0].__bases__
+                    if not find:  # Not the good type
+                        raise CheckTypeError(
+                            "For "
+                            + var_name
+                            + " : "
+                            + expect_type
+                            + " expected, "
+                            + type_value
+                            + " given"
+                        )
     elif "." in expect_type:
         # Imported type
         module_to_import = import_module(expect_type[: expect_type.rfind(".")])