Fix tests

materialsproject · Feb 16, 2024 · 3bc3ac8 · 3bc3ac8
1 parent 887904b
commit 3bc3ac8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/materials/test_eos.py b/tests/materials/test_eos.py
@@ -25,9 +25,10 @@ def rester():
 
 sub_doc_fields = []  # type: list
 
-alt_name_dict = {}  # type: dict
+alt_name_dict = {"material_ids": "material_id"}  # type: dict
 
-custom_field_tests = {}  # type: dict
+custom_field_tests = {  # type: dict
+    "material_ids": ["mp-149"]}
 
 
 @pytest.mark.skipif(os.getenv("MP_API_KEY", None) is None, reason="No API key found.")

diff --git a/tests/materials/test_grain_boundary.py b/tests/materials/test_grain_boundary.py
@@ -24,7 +24,7 @@ def rester():
 
 sub_doc_fields = []  # type: list
 
-alt_name_dict = {"material_ids": "task_id", "separation_energy": "w_sep"}  # type: dict
+alt_name_dict = {"material_ids": "material_id", "separation_energy": "w_sep"}  # type: dict
 
 custom_field_tests = {
     "material_ids": ["mp-81"],