From 3df1c0af382c6c04370b879c7168b46d833c761f Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Fri, 3 Dec 2021 16:53:55 +0100 Subject: Day3 p2 --- conftest.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) (limited to 'conftest.py') 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): if 'assignment' in metafunc.fixturenames: packages = [ importlib.import_module(package.name) - for package - in walk_packages([dir_path]) - if package.name.startswith('day') + for package in walk_packages([dir_path]) ] assignments = [ - (getattr(package, f'Assignment{day(part)}'), package) + (getattr(package, f'Assignment{day(part)}', None), package) for package in packages for part in ['1', '2'] ] @@ -27,8 +25,7 @@ def pytest_generate_tests(metafunc: Metafunc): argnames=f'assignment', argvalues=[ Assignment(path=package.__path__[0]) - for (Assignment, package) - in assignments + for (Assignment, package) in assignments if Assignment is not None ], ids=lambda assignment: str(assignment) -- cgit v1.2.3