From 6df440f3cd8d1bc9d2c181f81fcff35b9c4c1388 Mon Sep 17 00:00:00 2001 From: Shyam D Date: Thu, 22 Oct 2020 09:27:31 -0700 Subject: [PATCH] test for materials builder --- tests/emmet-builders/test_materials.py | 34 ++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 tests/emmet-builders/test_materials.py diff --git a/tests/emmet-builders/test_materials.py b/tests/emmet-builders/test_materials.py new file mode 100644 index 0000000000..e91b03a5d5 --- /dev/null +++ b/tests/emmet-builders/test_materials.py @@ -0,0 +1,34 @@ +import pytest +from maggma.stores import JSONStore, MemoryStore + +from emmet.builders.vasp.task_validator import TaskValidator +from emmet.builders.vasp.materials import MaterialsBuilder + + +@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(tasks_store): + validation_store = MemoryStore() + builder = TaskValidator(tasks=tasks_store, task_validation=validation_store) + builder.run() + return validation_store + + +@pytest.fixture +def materials_store(): + return MemoryStore() + + +def test_materials_builder(tasks_store, validation_store, materials_store): + + builder = MaterialsBuilder( + tasks=tasks_store, task_validation=validation_store, materials=materials_store + ) + builder.run() + assert materials_store.count() == 1 + print(materials_store.query_one()) + assert materials_store.count({"deprecated": False}) == 1