diff options
Diffstat (limited to 'conftest.py')
| -rw-r--r-- | conftest.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/conftest.py b/conftest.py index 3bae1e2..4832011 100644 --- a/conftest.py +++ b/conftest.py | |||
| @@ -12,13 +12,11 @@ def pytest_generate_tests(metafunc: Metafunc): | |||
| 12 | if 'assignment' in metafunc.fixturenames: | 12 | if 'assignment' in metafunc.fixturenames: |
| 13 | packages = [ | 13 | packages = [ |
| 14 | importlib.import_module(package.name) | 14 | importlib.import_module(package.name) |
| 15 | for package | 15 | for package in walk_packages([dir_path]) |
| 16 | in walk_packages([dir_path]) | ||
| 17 | if package.name.startswith('day') | ||
| 18 | ] | 16 | ] |
| 19 | 17 | ||
| 20 | assignments = [ | 18 | assignments = [ |
| 21 | (getattr(package, f'Assignment{day(part)}'), package) | 19 | (getattr(package, f'Assignment{day(part)}', None), package) |
| 22 | for package in packages | 20 | for package in packages |
| 23 | for part in ['1', '2'] | 21 | for part in ['1', '2'] |
| 24 | ] | 22 | ] |
| @@ -27,8 +25,7 @@ def pytest_generate_tests(metafunc: Metafunc): | |||
| 27 | argnames=f'assignment', | 25 | argnames=f'assignment', |
| 28 | argvalues=[ | 26 | argvalues=[ |
| 29 | Assignment(path=package.__path__[0]) | 27 | Assignment(path=package.__path__[0]) |
| 30 | for (Assignment, package) | 28 | for (Assignment, package) in assignments |
| 31 | in assignments | ||
| 32 | if Assignment is not None | 29 | if Assignment is not None |
| 33 | ], | 30 | ], |
| 34 | ids=lambda assignment: str(assignment) | 31 | ids=lambda assignment: str(assignment) |
