Add Enum support - Githubissues

christianhelle commented 1 year ago

This implements #6

Example usage:

from enum import Enum

class Weekday(Enum):
    MONDAY = 1
    TUESDAY = 2
    WEDNESDAY = 3
    THURSDAY = 4
    FRIDAY = 5
    SATURDAY = 6
    SUNDAY = 7

from autofaker import autodata

@autodata(Weekday)
def test_create_enum_class_returns_not_none(sut):
    assert sut is not None

@autodata(Weekday)
def test_create_enum_class_returns_instance(sut):
    assert isinstance(sut, Weekday)

import unittest
from autofaker import Autodata

class AnonymousWeekdayTestCase(unittest.TestCase):

    def test_create_enum_class_returns_not_none(self):
        self.assertIsNotNone(Autodata.create(Weekday))

    def test_create_enum_class_returns_instance(self):
        self.assertIsInstance(Autodata.create(Weekday), Weekday)

christianhelle commented 1 year ago

Does this work for you @TheRobotCarlson ?