ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2025-11-24 11:55:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix aggregate.py

Browse Source
This commit is contained in:
Leonardus Chen 2025-10-25 13:11:27 +07:00
parent d72d07caf7
commit 9111f5d4f6
1 changed files with 23 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
tests/typing/aggregate.py
View File

@ -1,4 +1,5 @@
from dependency_injector import providers from dependency_injector import providers
from typing_extensions import assert_type, Any
class Animal: ... class Animal: ...
@ -8,29 +9,36 @@ class Cat(Animal): ...
# Test 1: to check Aggregate provider # Test 1: to check Aggregate provider
provider1: providers.Aggregate[str] = providers.Aggregate( provider1 = providers.Aggregate(
a=providers.Object("str1"), a=providers.Object("str1"),
b=providers.Object("str2"), b=providers.Object("str2"),
) )
provider_a_1: providers.Provider[str] = provider1.a provider_a_1 = provider1.a
provider_b_1: providers.Provider[str] = provider1.b provider_b_1: providers.Provider[str] = provider1.b
val1: str = provider1("a") val1 = provider1("a")
assert_type(provider1, providers.Aggregate[str])
assert_type(provider_a_1, providers.Provider[str])
assert_type(provider_b_1, providers.Provider[str])
assert_type(val1, str)
provider1_set_non_string_keys: providers.Aggregate[str] = providers.Aggregate() provider1_set_non_string_keys = providers.Aggregate[str]()
provider1_set_non_string_keys.set_providers({Cat: providers.Object("str")}) provider1_set_non_string_keys.set_providers({Cat: providers.Object("str")})
provider_set_non_string_1: providers.Provider[str] = ( provider_set_non_string_1 = provider1_set_non_string_keys.providers[Cat]
provider1_set_non_string_keys.providers[Cat] assert_type(provider_set_non_string_1, providers.Provider[str])
)
provider1_new_non_string_keys: providers.Aggregate[str] = providers.Aggregate(
# TODO: Change providers.Aggregate to accept Mapping? Then remove explicit typing here
provider1_new_non_string_keys = providers.Aggregate[str](
{Cat: providers.Object("str")}, {Cat: providers.Object("str")},
) )
factory_new_non_string_1: providers.Provider[str] = ( factory_new_non_string_1 = provider1_new_non_string_keys.providers[Cat]
provider1_new_non_string_keys.providers[Cat] assert_type(provider1_new_non_string_keys, providers.Aggregate[str])
) assert_type(factory_new_non_string_1, providers.Provider[str])
provider1_no_explicit_typing = providers.Aggregate(a=providers.Object("str")) provider1_no_explicit_typing = providers.Aggregate(a=providers.Object("str"))
provider1_no_explicit_typing_factory: providers.Provider[str] = ( provider1_no_explicit_typing_factory = provider1_no_explicit_typing.providers["a"]
provider1_no_explicit_typing.providers["a"] provider1_no_explicit_typing_object = provider1_no_explicit_typing("a")
)
provider1_no_explicit_typing_object: str = provider1_no_explicit_typing("a") assert_type(provider1_no_explicit_typing_factory, providers.Provider[str])
assert_type(provider1_no_explicit_typing_object, str)