DUnion merge

Author: Bogdan Kalashnikov

attrs_api/dynamic_typing.py

@@ -14,6 +14,9 @@ def __init__(self, t: Union[type, BaseType]):
     def __str__(self):
         return f"{self.__class__.__name__}[{self.type}]"
 
+    def __eq__(self, other):
+        return isinstance(other, self.__class__) and self.type == other.type
+
 
 class Many(BaseType):
     __slots__ = ["types"]
@@ -25,13 +28,21 @@ def __str__(self):
         items = ', '.join(map(str, self.types))
         return f"{self.__class__.__name__}[{items}]"
 
+    def __eq__(self, other):
+        return isinstance(other, self.__class__) and all(t1 == t2 for t1, t2 in zip(self.types, other.types))
+
 
 class DOptional(Single):
     pass
 
 
 class DUnion(Many):
-    pass
+    def __init__(self, *types: Union[type, BaseType]):
+        unique_types = []
+        for t in types:
+            if t not in unique_types:
+                unique_types.append(t)
+        super().__init__(*unique_types)
 
 
 class DTuple(Many):

attrs_api/generator.py

@@ -80,7 +80,10 @@ def _detect_type(self, value, convert_dict=True) -> META_TYPE:
                     t = self._detect_type(item, convert_dict)
                     types.append(t)
                 if len(types) > 1:
-                    return DList(DUnion(*types))
+                    union = DUnion(*types)
+                    if len(union.types) == 1:
+                        return DList(*union.types)
+                    return DList(union)
                 else:
                     return DList(*types)
             else:

testing_tools/data.py

@@ -29,6 +29,23 @@
     },
     {
         "null_or_bool": True,
-        "str_number": "8"
+        "str_number": "8",
+        "3d_array": [
+            [
+                [0, 1, 2],
+                [3, 4, 5],
+                [6, 7, 8]
+            ],
+            [
+                [0, 1, 2],
+                [3, 4, 5],
+                [6, 7, 8]
+            ],
+            [
+                [0, 1, 2],
+                [3, 4, 5],
+                [6, 7, 8]
+            ]
+        ]
     },
 ]