pymocker 0.1.5

Generate realistic, context-aware mock data automatically.

PyMocker

This library is experimental

PyMocker is a powerful Python library that extends Polyfactory to automatically generate realistic, context-aware mock data for your Python data models. Say goodbye to random strings! it's a drop-in solution for creating professional mockups and test data.

Installation

Install via pip:

pip install pymocker

Example:

Add the @mocker.mock() decorator to any Polyfactory class, and it will automatically generate realistic, contextual data.

...
class Person(BaseModel):
    FirstName:str= Field(max_length=8)
    EmailAddress:str= Field(max_length=20)
    CellPhoneNumber:str= Field(min_length=12,max_length=12)

# Polyfactory
class PersonFactory(ModelFactory[Person]):...

# Polyfactory + Mocker
mocker=Mocker()
@mocker.mock()
class MockerPersonFactory(ModelFactory[Person]):...
...

Polyfactory:
Person(FirstName='48a40717', EmailAddress='1a5a1a37', CellPhoneNumber='6185d0d7c109')
Polyfactory + Mocker:
Person(FirstName='Ashley', EmailAddress='[e-mail]', CellPhoneNumber='429-860-3379')

See examples for more!

Intelligent Field Matching

PyMocker's internal ranking and similarity algorithms use a number of techniques to match fields to methods, including cosine similarity. You can adjust the confidence threshold for this behavior:

...
mocker=Mocker()
Mocker.confidence_threshold = 0.75 #.5 by default. higher means the model must be more confident to match
@mocker.mock()
...

Adding and customizing Providers

By default, PyMocker will use a Faker instance as its sole method provider. Configure your faker instance and add custom classes by adding it directly to Mocker's provider_instances.

...
class SuperHeroProvider:
    @staticmethod
    def super_hero_name():
        return 'MockerMan'
class Hero(BaseModel):
    HeroName:str=Field(max_length=9)
    
custom_faker_mocker=Mocker()
custom_faker_mocker.Config.provider_instances = [SuperHeroProvider(), Faker(locale='en_us')] # order affects matches

@custom_faker_mocker.mock()
...

Hero(HeroName='MockerMan')

Intelligent Field Matching

PyMocker uses a number of matching rules to match methods to fields, including cosine similarity. Configure this behavior like so:

#Control the Confidence threshold of similarity matching, .5 by default
mocker.confidence_threshold = 0.75

Note: Cosine Similarity is not perfect, and at times, may produce undesired results. You can disable this behavior entirely by setting match_field_generation_on_cosine_similarity to False

mocker.match_field_generation_on_cosine_similarity = False
# a confidence threshold of 0 also disables the behavior
mocker.confidence_threshold = 0

When disabled, PyMocker still uses word segmentation to discover matches for you. If no method is found, PyMocker defaults to PolyFactory's behavior

Other configurable attributes:

• max_retries (int): The number of times a method will attempt to generate a constraint-fulfilling value. Higher values can impact performance. Defaults to 300.
• coerce_on_fail (bool): If True, attempts to coerce the value to match constraints if Faker generation fails. Defaults to True. When set to False, PyMocker will default to a PolyFactory generated value

Supported Model Types

PyMocker seamlessly integrates with all PolyFactory Factories, except for SQLAlchemy - there's currently an issue with pk/fk relationships, so your milage may vary

Contributing

I'm just one guy, so I'd love some help improving this library. This is very early stages, so any suggestions or changes are welcome.