-
Notifications
You must be signed in to change notification settings - Fork 62
/
test_structuremetadata.py
67 lines (46 loc) · 1.99 KB
/
test_structuremetadata.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import pytest
from pymatgen import Element, Lattice
from emmet.core.structure import StructureMetadata
from emmet.core.symmetry import CrystalSystem, SymmetryData
from emmet.stubs import Structure
@pytest.fixture
def structure():
test_latt = Lattice.cubic(3.0)
test_struc = Structure(lattice=test_latt, species=["Fe"], coords=[[0, 0, 0]])
return test_struc
def test_symmetry(structure):
symm_doc = SymmetryData.from_structure(structure)
assert symm_doc.number == 221
assert symm_doc.point_group == "m-3m"
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 == [Element.Fe]
assert meta_doc.nelements == 1
assert meta_doc.formula_pretty == "Fe"
assert meta_doc.formula_anonymous == "A"
assert meta_doc.chemsys == "Fe"
assert meta_doc.volume == 27.0
assert meta_doc.density == 3.4345483027509993
assert meta_doc.density_atomic == 27.0
def test_structure_metadata_fewer_fields(structure):
meta_doc = StructureMetadata.from_structure(
structure, fields=["nsites", "nelements", "volume"]
)
assert meta_doc.nsites == 1
assert meta_doc.nelements == 1
assert meta_doc.volume == 27.0
def test_composition(structure):
meta_doc = StructureMetadata.from_structure(structure)
comp_meta_doc = StructureMetadata.from_composition(structure.composition)
assert meta_doc.elements == comp_meta_doc.elements
assert meta_doc.nelements == comp_meta_doc.nelements
assert meta_doc.formula_pretty == comp_meta_doc.formula_pretty
assert meta_doc.formula_anonymous == comp_meta_doc.formula_anonymous
assert meta_doc.chemsys == comp_meta_doc.chemsys
def test_schema():
StructureMetadata.schema()