test: refactor the tests to move tests to correct places (#495)

Author: vkcku

tests/test_annotated_fields.py

@@ -1,7 +1,7 @@
 import datetime as dt
 from decimal import Decimal
 from enum import Enum
-from typing import Any, Dict, Generic, List, Tuple, TypeVar
+from typing import Any, Dict, Generic, List, Optional, Tuple, TypeVar, Union
 from uuid import UUID
 
 import attrs
@@ -219,3 +219,50 @@ class FooFactory(AttrsFactory[Foo]):
     foo = FooFactory.build()
 
     assert foo.default_field == 10
+
+
+def test_union_types() -> None:
+    @define
+    class A:
+        a: Union[List[str], List[int]]
+        b: Union[str, List[int]]
+        c: List[Union[Tuple[int, int], Tuple[str, int]]]
+
+    AFactory = AttrsFactory.create_factory(A)
+
+    assert AFactory.build()
+
+
+def test_collection_unions_with_models() -> None:
+    @define
+    class A:
+        a: int
+
+    @define
+    class B:
+        a: str
+
+    @define
+    class C:
+        a: Union[List[A], List[B]]
+        b: List[Union[A, B]]
+
+    CFactory = AttrsFactory.create_factory(C)
+
+    assert CFactory.build()
+
+
+@pytest.mark.parametrize("allow_none", (True, False))
+def test_optional_type(allow_none: bool) -> None:
+    @define
+    class A:
+        a: Union[str, None]
+        b: Optional[str]
+        c: Optional[Union[str, int, List[int]]]
+
+    class AFactory(AttrsFactory[A]):
+        __model__ = A
+
+        __allow_none_optionals__ = allow_none
+
+    assert AFactory.build()

tests/test_attrs_factory.py

@@ -1,8 +1,6 @@
 from uuid import uuid4
 
-import pytest
-
-from pydantic import VERSION, BaseModel, Field, ValidationError
+from pydantic import BaseModel
 
 from polyfactory.factories.pydantic_factory import ModelFactory
 from tests.models import PersonFactoryWithDefaults, Pet, PetFactory
@@ -62,61 +60,6 @@ def test_builds_batch() -> None:
         assert isinstance(result.age, float)
 
 
-def test_factory_use_construct() -> None:
-    # factory should pass values without validation
-    invalid_age = "non_valid_age"
-    non_validated_pet = PetFactory.build(factory_use_construct=True, age=invalid_age)
-    assert non_validated_pet.age == invalid_age
-
-    with pytest.raises(ValidationError):
-        PetFactory.build(age=invalid_age)
-
-    with pytest.raises(ValidationError):
-        PetFactory.build(age=invalid_age)
-
-
-@pytest.mark.skipif(VERSION.startswith("2"), reason="pydantic 1 only test")
-def test_build_instance_by_field_alias_with_allow_population_by_field_name_flag_pydantic_v1() -> None:
-    class MyModel(BaseModel):
-        aliased_field: str = Field(..., alias="special_field")
-
-        class Config:
-            allow_population_by_field_name = True
-
-    class MyFactory(ModelFactory):
-        __model__ = MyModel
-
-    instance = MyFactory.build(aliased_field="some")
-    assert instance.aliased_field == "some"
-
-
-@pytest.mark.skipif(VERSION.startswith("1"), reason="pydantic 2 only test")
-def test_build_instance_by_field_alias_with_populate_by_name_flag_pydantic_v2() -> None:
-    class MyModel(BaseModel):
-        model_config = {"populate_by_name": True}
-        aliased_field: str = Field(..., alias="special_field")
-
-    class MyFactory(ModelFactory):
-        __model__ = MyModel
-
-    instance = MyFactory.build(aliased_field="some")
-    assert instance.aliased_field == "some"
-
-
-def test_build_instance_by_field_name_with_allow_population_by_field_name_flag() -> None:
-    class MyModel(BaseModel):
-        aliased_field: str = Field(..., alias="special_field")
-
-        class Config:
-            allow_population_by_field_name = True
-
-    class MyFactory(ModelFactory):
-        __model__ = MyModel
-
-    instance = MyFactory.build(special_field="some")
-    assert instance.aliased_field == "some"
-
-
 def test_build_model_with_fields_named_like_factory_fields() -> None:
     class C(BaseModel):
         batch: int

tests/test_setting_default_factories.py renamed to tests/test_auto_registration.py

@@ -21,7 +21,7 @@
 
 import pytest
 
-from pydantic import VERSION, BaseModel
+from pydantic import BaseModel
 
 from polyfactory.exceptions import ParameterException
 from polyfactory.factories import DataclassFactory
@@ -81,20 +81,6 @@ class MyFactory(ModelFactory):
     assert result.person_list[0].pets
 
 
-@pytest.mark.skipif(VERSION.startswith("2"), reason="pydantic 1 only test")
-def test_handles_complex_typing_with_custom_root_type() -> None:
-    class MyModel(BaseModel):
-        __root__: List[int]
-
-    class MyFactory(ModelFactory[MyModel]):
-        __model__ = MyModel
-
-    result = MyFactory.build()
-
-    assert result.__root__
-    assert isinstance(result.__root__, list)
-
-
 def test_raises_for_user_defined_types() -> None:
     class MyClass:
         def __init__(self, value: int) -> None: