From 903d4df00b18df8ec12a114e7ed17d6b0b7fd2d0 Mon Sep 17 00:00:00 2001 From: Shyam D Date: Wed, 18 Nov 2020 09:55:05 -0800 Subject: [PATCH] fix breaks in tests from ValueEnum --- tests/emmet-core/test_structuremetadata.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/emmet-core/test_structuremetadata.py b/tests/emmet-core/test_structuremetadata.py index fee8efe4f7..6735c957b6 100644 --- a/tests/emmet-core/test_structuremetadata.py +++ b/tests/emmet-core/test_structuremetadata.py @@ -1,5 +1,5 @@ import pytest -from pymatgen import Lattice +from pymatgen import Lattice, Element from emmet.core.structure import StructureMetadata from emmet.core.symmetry import CrystalSystem, SymmetryData @@ -22,13 +22,16 @@ def test_symmetry(structure): assert symm_doc.symbol == "Pm-3m" assert symm_doc.crystal_system == CrystalSystem.cubic + assert symm_doc.dict()["crystal_system"] == CrystalSystem.cubic + assert str(symm_doc.dict()["crystal_system"]) == "Cubic" + def test_structure_metadata(structure): meta_doc = StructureMetadata.from_structure(structure) assert meta_doc.nsites == 1 - assert meta_doc.elements == ["Fe"] + assert meta_doc.elements == [Element.Fe] assert meta_doc.nelements == 1 assert meta_doc.formula_pretty == "Fe" assert meta_doc.formula_anonymous == "A"