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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
|
"""Shared test functions and attributes."""
import atexit
import os
from contextlib import ExitStack
from pathlib import Path
from schema_salad.utils import as_file, files
def get_path(filename: str) -> Path:
"""Get the file path for a given schema file name.
It is able to find file names in the ``schema_salad`` namespace, but
also able to load schema files from the ``tests`` directory.
"""
filename = os.path.normpath(filename) # normalizing path depending on OS
# or else it will cause problem when joining path
filepath = None
try:
file_manager = ExitStack()
atexit.register(file_manager.close)
traversable = files("schema-salad") / filename
filepath = file_manager.enter_context(as_file(traversable))
except ModuleNotFoundError:
pass
if not filepath or not filepath.is_file():
# First try to load it from the local directory, probably ``./tests/``.
filepath = Path(os.path.dirname(__file__)) / filename
if not filepath.is_file():
# If that didn't work, then default to tests/../${filename},
# note that we return the parent as it is expected that __file__
# is a test file.
filepath = Path(os.path.dirname(__file__)) / ".." / filename
return filepath.resolve()
def get_data(filename: str) -> str:
"""Get the file path for a given schema file name.
It is able to find file names in the ``schema_salad`` namespace, but
also able to load schema files from the ``tests`` directory.
"""
return str(get_path(filename))
def get_data_uri(resource_path: str) -> str:
"""Get the file URI for tests."""
return get_path(resource_path).as_uri()
# Schemas used in tests
cwl_file_uri = get_data_uri("tests/test_schema/CommonWorkflowLanguage.yml")
metaschema_file_uri = get_data_uri("metaschema/metaschema.yml")
basket_file_uri = get_data_uri("tests/basket_schema.yml")
|