diff --git a/tests/emmet-builders/test_vasp.py b/tests/emmet-builders/test_vasp.py new file mode 100644 index 0000000000..4581efefd8 --- /dev/null +++ b/tests/emmet-builders/test_vasp.py @@ -0,0 +1,23 @@ +import pytest +from maggma.stores import JSONStore, MemoryStore +from emmet.builders.vasp.task_validator import TaskValidator + + +intermediate_stores = ["validation"] + + +@pytest.fixture(scope="session") +def tasks_store(test_dir): + return JSONStore(test_dir / "test_si_tasks.json.gz") + + +@pytest.fixture(scope="session") +def validation_store(): + return MemoryStore() + + +def test_validator(tasks_store, validation_store): + builder = TaskValidator(tasks=tasks_store, task_validation=validation_store) + builder.run() + assert validation_store.count() == tasks_store.count() + assert validation_store.count({"valid": True}) == tasks_store.count()