test_vasp.py

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()