Skip to content

Commit

Permalink
test generic forward reference
Browse files Browse the repository at this point in the history
  • Loading branch information
avlonder committed Jan 19, 2025
1 parent b982236 commit 3eb8f93
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 2 deletions.
26 changes: 25 additions & 1 deletion tests/core/test_forward_reference.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
from dataclasses import dataclass
from typing import List, Optional
from typing import Generic, List, Optional, TypeVar
from dacite import from_dict

T = TypeVar("T")


@dataclass
class Person:
Expand All @@ -15,6 +17,18 @@ class Club:
members: List["Person"]


@dataclass
class Employee:
name: str


@dataclass
class Team(Generic[T]):
name: str
members: Optional[List[T]] = None
subteams: Optional[List["Team[T]"]] = None


def test_self_reference():
data = {"name": "John Doe", "children": [{"name": "Jane Doe"}]}

Expand All @@ -29,3 +43,13 @@ def test_other_reference():
result = from_dict(Club, data)

assert result == Club(name="FooBar", members=[Person(name="John Doe", children=[Person(name="Jane Doe")])])


def test_generic_self_reference():
data = {"name": "foo", "members": [{"name": "John"}], "subteams": [{"name": "bar", "members": [{"name": "Jane"}]}]}

result = from_dict(Team[Employee], data)

assert result == Team(
name="foo", members=[Employee(name="John")], subteams=[Team(name="bar", members=[Employee(name="Jane")])]
)
2 changes: 1 addition & 1 deletion tests/core/test_generics.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def test_multi_generic():
assert result == A(x=X(a="foo"), y=[1, 2, 3])


def test_inherited_generic():
def test_generic_parent():
@dataclass
class B(A[X, int]):
z: str
Expand Down

0 comments on commit 3eb8f93

Please # to comment.