import pytest
from pytest_examples import CodeExample, EvalExample

from tests.docs._concept_groups import collect_examples, core_concept_files

code_examples = collect_examples(core_concept_files())


@pytest.mark.parametrize("example", code_examples, ids=str)
def test_format_concepts_core(example: CodeExample, eval_example: EvalExample):
    if eval_example.update_examples:
        eval_example.format(example)
        eval_example.run_print_update(example)
    else:
        eval_example.lint(example)
        eval_example.run(example)