1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
|
# Copyright: See the LICENSE file.
import dataclasses
import unittest
import factory
@dataclasses.dataclass
class User:
name: str
email: str
id: int
class TypingTests(unittest.TestCase):
def test_simple_factory(self) -> None:
class UserFactory(factory.Factory[User]):
name = "John Doe"
email = "john.doe@example.org"
id = 42
class Meta:
model = User
result: User
result = UserFactory.build()
result = UserFactory.create()
self.assertEqual(result.name, "John Doe")
|