blob: 1c0a8e4c2f7fe2fcb07969a1048e2e705b777d16 (
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
33
34
35
36
|
# -*- coding: utf-8 -*-
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
and hasattr(package, "__path__")
and Assignment.example_result != NotImplemented
],
ids=lambda assignment: str(assignment),
)
|
