mirror of
https://github.com/p2p-ld/numpydantic.git
synced 2024-09-19 23:44:28 +00:00
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
import shutil
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from linkml_runtime.linkml_model import ClassDefinition, SlotDefinition
|
|
|
|
DATA_DIR = Path(__file__).parent / "data"
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def tmp_output_dir() -> Path:
|
|
path = Path(__file__).parent.resolve() / "__tmp__"
|
|
if path.exists():
|
|
shutil.rmtree(str(path))
|
|
path.mkdir()
|
|
|
|
return path
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def tmp_output_dir_func(tmp_output_dir) -> Path:
|
|
"""
|
|
tmp output dir that gets cleared between every function
|
|
cleans at the start rather than at cleanup in case the output is to be inspected
|
|
"""
|
|
subpath = tmp_output_dir / "__tmpfunc__"
|
|
if subpath.exists():
|
|
shutil.rmtree(str(subpath))
|
|
subpath.mkdir()
|
|
return subpath
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def tmp_output_dir_mod(tmp_output_dir) -> Path:
|
|
"""
|
|
tmp output dir that gets cleared between every function
|
|
cleans at the start rather than at cleanup in case the output is to be inspected
|
|
"""
|
|
subpath = tmp_output_dir / "__tmpmod__"
|
|
if subpath.exists():
|
|
shutil.rmtree(str(subpath))
|
|
subpath.mkdir()
|
|
return subpath
|
|
|
|
|
|
@pytest.fixture()
|
|
def nwb_linkml_array() -> tuple[ClassDefinition, str]:
|
|
classdef = ClassDefinition(
|
|
name="NWB_Linkml Array",
|
|
description="Main class's array",
|
|
is_a="Arraylike",
|
|
attributes=[
|
|
SlotDefinition(name="x", range="numeric", required=True),
|
|
SlotDefinition(name="y", range="numeric", required=True),
|
|
SlotDefinition(
|
|
name="z",
|
|
range="numeric",
|
|
required=False,
|
|
maximum_cardinality=3,
|
|
minimum_cardinality=3,
|
|
),
|
|
SlotDefinition(
|
|
name="a",
|
|
range="numeric",
|
|
required=False,
|
|
minimum_cardinality=4,
|
|
maximum_cardinality=4,
|
|
),
|
|
],
|
|
)
|
|
generated = """Union[
|
|
NDArray[Shape["* x, * y"], Number],
|
|
NDArray[Shape["* x, * y, 3 z"], Number],
|
|
NDArray[Shape["* x, * y, 3 z, 4 a"], Number]
|
|
]"""
|
|
return classdef, generated
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def patch_slotarray():
|
|
import sys
|
|
|
|
# modname = "linkml_runtime.linkml_model.meta"
|
|
# if modname in sys.modules:
|
|
# del sys.modules[modname]
|
|
|
|
from numpydantic.linkml.slotarray import patch_linkml
|
|
import linkml_runtime
|
|
from linkml_runtime.linkml_model.meta import SlotDefinition
|
|
|
|
original_slot = SlotDefinition
|
|
|
|
yield patch_linkml()
|
|
|
|
linkml_runtime.linkml_model.SlotDefinition = original_slot
|
|
|
|
|
|
@pytest.fixture()
|
|
def slotarray_schemaview(patch_slotarray):
|
|
from linkml_runtime.utils.schemaview import SchemaView
|
|
|
|
return SchemaView(DATA_DIR / "slotarray.yaml")
|