Fix test_generate_class_file

Author: T4rk1n

tests/unit/development/metadata_test.py

@@ -1,13 +1,21 @@
 # AUTO GENERATED FILE - DO NOT EDIT
 
 import typing  # noqa: F401
-import numbers # noqa: F401
 from typing_extensions import TypedDict, NotRequired, Literal # noqa: F401
-from dash.development.base_component import Component
-try:
-    from dash.development.base_component import ComponentType # noqa: F401
-except ImportError:
-    ComponentType = typing.TypeVar("ComponentType", bound=Component)
+from dash.development.base_component import Component, _explicitize_args
+
+ComponentType = typing.Union[
+    str,
+    int,
+    float,
+    Component,
+    None,
+    typing.Sequence[typing.Union[str, int, float, Component, None]],
+]
+
+NumberType = typing.Union[
+    typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex
+]
 
 
 class Table(Component):
@@ -109,7 +117,7 @@ class Table(Component):
         "OptionalObjectWithExactAndNestedDescription",
             {
             "color": NotRequired[str],
-            "fontSize": NotRequired[typing.Union[typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex]],
+            "fontSize": NotRequired[NumberType],
             "figure": NotRequired["OptionalObjectWithExactAndNestedDescriptionFigure"]
         }
     )
@@ -126,30 +134,29 @@ class Table(Component):
         "OptionalObjectWithShapeAndNestedDescription",
             {
             "color": NotRequired[str],
-            "fontSize": NotRequired[typing.Union[typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex]],
+            "fontSize": NotRequired[NumberType],
             "figure": NotRequired["OptionalObjectWithShapeAndNestedDescriptionFigure"]
         }
     )
 
-    _explicitize_dash_init = True
 
     def __init__(
         self,
-        children: typing.Optional[typing.Union[str, int, float, ComponentType, typing.Sequence[typing.Union[str, int, float, ComponentType]]]] = None,
+        children: typing.Optional[ComponentType] = None,
         optionalArray: typing.Optional[typing.Sequence] = None,
         optionalBool: typing.Optional[bool] = None,
         optionalFunc: typing.Optional[typing.Any] = None,
-        optionalNumber: typing.Optional[typing.Union[typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex]] = None,
+        optionalNumber: typing.Optional[NumberType] = None,
         optionalObject: typing.Optional[dict] = None,
         optionalString: typing.Optional[str] = None,
         optionalSymbol: typing.Optional[typing.Any] = None,
-        optionalNode: typing.Optional[typing.Union[str, int, float, ComponentType, typing.Sequence[typing.Union[str, int, float, ComponentType]]]] = None,
-        optionalElement: typing.Optional[ComponentType] = None,
+        optionalNode: typing.Optional[ComponentType] = None,
+        optionalElement: typing.Optional[Component] = None,
         optionalMessage: typing.Optional[typing.Any] = None,
         optionalEnum: typing.Optional[Literal["News", "Photos"]] = None,
-        optionalUnion: typing.Optional[typing.Union[str, typing.Union[typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex], typing.Any]] = None,
-        optionalArrayOf: typing.Optional[typing.Sequence[typing.Union[typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex]]] = None,
-        optionalObjectOf: typing.Optional[typing.Dict[typing.Union[str, float, int], typing.Union[typing.SupportsFloat, typing.SupportsInt, typing.SupportsComplex]]] = None,
+        optionalUnion: typing.Optional[typing.Union[str, NumberType, typing.Any]] = None,
+        optionalArrayOf: typing.Optional[typing.Sequence[NumberType]] = None,
+        optionalObjectOf: typing.Optional[typing.Dict[typing.Union[str, float, int], NumberType]] = None,
         optionalObjectWithExactAndNestedDescription: typing.Optional["OptionalObjectWithExactAndNestedDescription"] = None,
         optionalObjectWithShapeAndNestedDescription: typing.Optional["OptionalObjectWithShapeAndNestedDescription"] = None,
         optionalAny: typing.Optional[typing.Any] = None,
@@ -168,3 +175,5 @@ def __init__(
         args = {k: _locals[k] for k in _explicit_args if k != 'children'}
 
         super(Table, self).__init__(children=children, **args)
+
+setattr(Table, "__init__", _explicitize_args(Table.__init__))

tests/unit/development/test_generate_class_file.py

@@ -33,7 +33,7 @@ def expected_class_string():
 
 @pytest.fixture
 def component_class_string(make_component_dir):
-    return import_string + generate_class_string(
+    return import_string.format(custom_imports="") + generate_class_string(
         typename="Table",
         props=make_component_dir["props"],
         description=make_component_dir["description"],