blob: 4832011fa45695160f6f0c6b1548e11391f22208 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
import importlib
import os
from pkgutil import walk_packages
from _pytest.python import Metafunc
from aoc.__main__ import day
dir_path = os.path.dirname(os.path.realpath(__file__))
def pytest_generate_tests(metafunc: Metafunc):
if 'assignment' in metafunc.fixturenames:
packages = [
importlib.import_module(package.name)
for package in walk_packages([dir_path])
]
assignments = [
(getattr(package, f'Assignment{day(part)}', None), package)
for package in packages
for part in ['1', '2']
]
metafunc.parametrize(
argnames=f'assignment',
argvalues=[
Assignment(path=package.__path__[0])
for (Assignment, package) in assignments
if Assignment is not None
],
ids=lambda assignment: str(assignment)
)
|
