ensure run_type is tested in builders and doc model

materialsproject · Jun 24, 2021 · 3c419b7 · 3c419b7
1 parent 2c2b5a5
commit 3c419b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/emmet-builders/test_vasp.py b/tests/emmet-builders/test_vasp.py
@@ -21,3 +21,4 @@ def test_validator(tasks_store, validation_store):
     builder.run()
     assert validation_store.count() == tasks_store.count()
     assert validation_store.count({"valid": True}) == tasks_store.count()
+    assert all(list(d["run_type"] == "GGA" for d in validation_store.query()))
diff --git a/tests/emmet-core/vasp/test_vasp.py b/tests/emmet-core/vasp/test_vasp.py
@@ -69,4 +69,5 @@ def task_ldau(test_dir):
 
 
 def test_ldau(task_ldau):
+    assert task_ldau.run_type == RunType.GGA_U
     assert ValidationDoc.from_task_doc(task_ldau).valid is False