Skip to content

Commit

Permalink
update nomad exmaple tests
Browse files Browse the repository at this point in the history
  • Loading branch information
lukaspie committed Nov 26, 2024
1 parent c6120e5 commit 2724aa4
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 15 deletions.
1 change: 1 addition & 0 deletions src/pynxtools_apm/nomad/entrypoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,5 +36,6 @@
and keep the size of your NOMAD installation small. Once started, we recommend changing the respective
input file in the NOMAD Oasis ELN to run the example with your own datasets.
""",
plugin_package="pynxtools_apm",
resources=["nomad/examples/*"],
)
36 changes: 21 additions & 15 deletions tests/test_nomad_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,38 +38,44 @@
from pynxtools_apm.nomad.entrypoints import apm_example


EXAMPLE_PATH = os.path.join(
os.path.dirname(__file__),
"..",
"src",
"pynxtools_apm",
"nomad",
"examples",
)


@pytest.mark.parametrize(
"mainfile",
get_file_parameter(
os.path.join(
os.path.dirname(__file__),
"..",
"src",
"pynxtools_apm",
"nomad",
"examples",
)
),
get_file_parameter(EXAMPLE_PATH),
)
def test_parse_nomad_examples(mainfile):
"""Test if NOMAD examples work."""
archive_dict = parse_nomad_examples(mainfile)


@pytest.mark.parametrize(
("entrypoint", "expected_local_path"),
("entrypoint", "example_path"),
[
pytest.param(
apm_example,
"examples/data/uploads/apm.zip",
EXAMPLE_PATH,
id="apm_example",
),
],
)
def test_nomad_example_upload_entry_point_valid(entrypoint, expected_local_path):
def test_example_upload_entry_point_valid(entrypoint, example_path):
"""Test if NOMAD ExampleUploadEntryPoint works."""
expected_upload_files = []
for dirpath, dirnames, filenames in os.walk(example_path):
for filename in filenames:
file_path = os.path.relpath(os.path.join(dirpath, filename), example_path)
expected_upload_files.append(file_path)

example_upload_entry_point_valid(
entrypoint=entrypoint,
plugin_package="pynxtools-apm",
expected_local_path=expected_local_path,
expected_upload_files=expected_upload_files,
)

0 comments on commit 2724aa4

Please sign in to comment.