test_data.py

"""Data test."""
import os
import glob
import pytest
from pathlib import Path

import include_access_model.datamodel.include_access_model
from linkml_runtime.loaders import yaml_loader

DATA_DIR_VALID = Path(__file__).parent / "data" / "valid"
DATA_DIR_INVALID = Path(__file__).parent / "data" / "invalid"

VALID_EXAMPLE_FILES = glob.glob(os.path.join(DATA_DIR_VALID, '*.yaml'))
INVALID_EXAMPLE_FILES = glob.glob(os.path.join(DATA_DIR_INVALID, '*.yaml'))


@pytest.mark.parametrize("filepath", VALID_EXAMPLE_FILES)
def test_valid_data_files(filepath):
    """Test loading of all valid data files."""
    target_class_name = Path(filepath).stem.split("-")[0]
    tgt_class = getattr(
        include_access_model.datamodel.include_access_model,
        target_class_name,
    )
    obj = yaml_loader.load(filepath, target_class=tgt_class)
    assert obj

"""This may not work for every expected error type- it uses ValueError"""
@pytest.mark.parametrize("filepath", INVALID_EXAMPLE_FILES)
def test_valid_data_files(filepath):
    """Test loading of all valid data files."""
    target_class_name = Path(filepath).stem.split("-")[0]
    tgt_class = getattr(
        include_access_model.datamodel.include_access_model,
        target_class_name,
    )
    with pytest.raises(ValueError):
        obj = yaml_loader.load(filepath, target_class=tgt_class)