from forge import Forge with Forge.define('color') as f: f.name = 'Red'
def setUp(self): Forge.configure(models='forge.test.support.models') if 'user' not in Forge._registry: Forge.define('user', name='Matte', age=25)
def should_support_with_context_format(self): with Forge.define('pet') as p: p.breed = 'Husky' assert Forge._registry['pet'] == {'breed': 'Husky'}
def should_create_associations(self): Forge.define('car', owner=Forge.build('user')) assert Forge.build('car').owner.name == 'Matte'
def should_not_allow_duplicate_factory_names(self): with self.assertRaises(DuplicateFactoryError): Forge.define('user', color='red')