def test_make_defect_entry(defect_entry):
relaxed_coords = \
[[0.25, 0.0, 0.0], [0.5, 0.5, 0.0], [0.5, 0.0, 0.5], [0.0, 0.5, 0.5],
[0.0, 0.0, 0.51], [0.0, 0.51, 0.0], [0.5, 0.0, 0.0], [0.5, 0.5, 0.5]]
relaxed_defect = IStructure(Lattice.cubic(10.0),
["Li"] + ["H"] * 3 + ["He"] * 4,
relaxed_coords)
unrelaxed_coords = \
[[0.25, 0.0, 0.0], [0.5, 0.5, 0.0], [0.5, 0.0, 0.5], [0.0, 0.5, 0.5],
[0.0, 0.0, 0.5], [0.0, 0.5, 0.0], [0.5, 0.0, 0.0], [0.5, 0.5, 0.5]]
unrelaxed_defect = IStructure(Lattice.cubic(10.0),
["Li"] + ["H"] * 3 + ["He"] * 4,
unrelaxed_coords)
actual = make_defect_entry(name="Va_O1",
charge=1,
perfect_structure=perfect,
defect_structure=relaxed_defect)
expected = DefectEntry(name="Va_O1",
charge=1,
structure=unrelaxed_defect,
perturbed_structure=None,
site_symmetry="4mm",
defect_center=(0.125, 0.0, 0.0))
assert actual == expected
def test_make_calc_summary(mocker):
defect_entry = mocker.Mock(spec=DefectEntry, autospec=True)
defect_entry.name = "Va_O1"
defect_entry.charge = 1
defect_entry.full_name = "Va_O1_1"
calc_results = mocker.Mock(spec=CalcResults, autospec=True)
calc_results.structure = \
IStructure(Lattice.cubic(1.0), ["Mg"], [[0.0]*3])
calc_results.energy = 10.0 + defaults.abs_strange_energy - 0.1
structure_info = mocker.Mock(spec=DefectStructureInfo, autospec=True)
p_calc_results = mocker.Mock(spec=CalcResults, autospec=True)
p_calc_results.structure = \
IStructure(Lattice.cubic(1.0), ["Mg", "O"], [[0.0]*3]*2)
p_calc_results.energy = 10.0
actual = make_calc_summary(calc_set=[(calc_results, defect_entry,
structure_info)],
p_calc_results=p_calc_results)
single_summary = SingleCalcSummary(
charge=1,
atom_io={"O": -1},
electronic_conv=calc_results.electronic_conv,
ionic_conv=calc_results.ionic_conv,
is_energy_strange=False,
same_config_from_init=structure_info.same_config_from_init,
defect_type=str(structure_info.defect_type),
symm_relation=str(structure_info.symm_relation))
expected = CalcSummary({"Va_O1_1": single_summary})
assert actual == expected
def test_make_defect_energy_info(mocker):
defect_entry = mocker.Mock(DefectEntry, autospec=True)
defect_entry.name = "Va_Mg1"
defect_entry.charge = -1
calc_results = mocker.Mock(CalcResults, autospec=True)
calc_results.structure = IStructure(Lattice.cubic(1.0), ["O"], [[0.0]*3])
calc_results.energy = 10.0
calc_results.electronic_conv = False
correction = mocker.Mock(Correction, autospec=True)
correction.correction_dict = {"a": 10.0}
p_calc_results = mocker.Mock(CalcResults, autospec=True)
p_calc_results.structure = IStructure(Lattice.cubic(1.0),
["Mg", "O"], [[0.0]*3]*2)
p_calc_results.energy = 1.0
standard_energies = StandardEnergies({"Mg": 10.0, "O": 20.0})
unitcell = mocker.Mock()
unitcell.vbm = 100.0
actual = make_defect_energy_info(defect_entry, calc_results, correction,
p_calc_results, standard_energies, unitcell)
energy = DefectEnergy(formation_energy=10.0 - 1.0 + 10 - 100.0,
energy_corrections={"a": 10.0},
is_shallow=None)
expected = DefectEnergyInfo(name="Va_Mg1", charge=-1,
atom_io={"Mg": -1}, defect_energy=energy)
assert actual == expected
def structure_comparator():
cu2o_perfect = IStructure(
Lattice.cubic(5),
species=["Cu"] * 4 + ["O"] * 2,
coords=[
[0.25, 0.25, 0.25], # removed
[0.25, 0.74, 0.74], # removed
[0.75, 0.75, 0.25],
[0.75, 0.25, 0.75],
[0, 0, 0],
[0.5, 0.5, 0.5]
])
cu2o_defect = IStructure(
Lattice.cubic(5),
species=["Cu"] * 3 + ["O"] * 2 + ["H"],
coords=[
[0.25, 0.5, 0.5], # inserted
[0.76, 0.73, 0.24],
[0.75, 0.25, 0.73],
[0.05, 0.95, 0],
[0.5, 0.5, 0.5],
[0.25] * 3
]) # inserted
return DefectStructureComparator(defect_structure=cu2o_defect,
perfect_structure=cu2o_perfect)
def structure_comparator_periodic_issue():
cu2o_perfect = IStructure(Lattice.cubic(5),
species=["Cu"] * 4 + ["O"] * 2,
coords=[[0.25, 0.25, 0.25], [0.25, 0.75, 0.75],
[0.75, 0.75, 0.25], [0.75, 0.25, 0.75],
[0, 0, 0], [0.5, 0.5, 0.5]])
# defect center is ([1.0, 1.0, 1.0] + [0.99, 0.99, 0.99]) / 2 = [0.995]*3
cu2o_defect = IStructure(Lattice.cubic(5),
species=["Cu"] * 4 + ["O"] + ["H"],
coords=[[0.25, 0.25, 0.25], [0.25, 0.75, 0.75],
[0.75, 0.75, 0.25], [0.75, 0.25, 0.75],
[0.5, 0.5, 0.5], [0.99, 0.99, 0.99]])
return DefectStructureComparator(defect_structure=cu2o_defect,
perfect_structure=cu2o_perfect)
def complex_monoclinic():
lattice = Lattice.monoclinic(3, 4, 5, 100)
coords = [[0.0, 0.0, 0.0], [0.1, 0.0, 0.0], [0.9, 0.0, 0.0],
[0.2, 0.0, 0.0], [0.8, 0.0, 0.0]]
return IStructure(lattice=lattice,
species=["H", "He", "He", "He", "He"],
coords=coords)
def simple_cubic_2x2x2():
lattice = Lattice.cubic(2.0)
coords = [[0.0, 0.0, 0.0], [0.5, 0.5, 0.0], [0.5, 0.0,
0.5], [0.0, 0.5, 0.5],
[0.0, 0.0, 0.5], [0.0, 0.5, 0.0], [0.5, 0.0, 0.0],
[0.5, 0.5, 0.5]]
return IStructure(lattice=lattice, species=["H"] * 8, coords=coords)
def calc_results():
return CalcResults(structure=IStructure(Lattice.cubic(1.0), ["H"],
[[0.0] * 3]),
energy=1.0,
magnetization=0.0,
potentials=[0.0],
electronic_conv=False,
ionic_conv=False)
def test_make_site_diff_wo_diff():
s1 = IStructure(Lattice.cubic(10), species=["H"], coords=[[0] * 3])
s2 = s1.copy()
structure_comparator = DefectStructureComparator(s1, s2)
site_diff = structure_comparator.make_site_diff()
assert site_diff == SiteDiff([], [], [], [])
assert site_diff.is_no_diff
s1 = IStructure(Lattice.cubic(10),
species=["H", "He"],
coords=[[0] * 3, [0.5] * 3])
s2 = IStructure(Lattice.cubic(10), species=["H"], coords=[[0] * 3])
structure_comparator = DefectStructureComparator(defect_structure=s2,
perfect_structure=s1)
site_diff = structure_comparator.make_site_diff()
assert site_diff == SiteDiff([(1, "He", (0.5, 0.5, 0.5))], [], [], [])
assert site_diff.is_no_diff is False
def test_make_efnv_correction(mocker):
mock_perfect = mocker.Mock(spec=CalcResults, autospec=True)
mock_defect = mocker.Mock(spec=CalcResults, autospec=True)
mock_perfect.structure = IStructure(Lattice.cubic(10),
species=["H"] + ["He"] * 3 + ["Li"],
coords=[[0, 0, 0], [1 / 2, 1 / 2, 0],
[1 / 2, 0, 1 / 2],
[0, 1 / 2, 1 / 2],
[0, 0, 1 / 2]])
mock_defect.structure = IStructure(Lattice.cubic(10),
species=["He"] * 3 + ["Li"],
coords=[[1 / 2, 1 / 2, 0],
[1 / 2, 0, 1 / 2],
[0, 1 / 2, 1 / 2],
[0, 0, 1 / 2]])
mock_perfect.potentials = [3.0, 4.0, 5.0, 6.0, 7.0]
mock_defect.potentials = [14.0, 25.0, 36.0, 47.0]
mock_ewald = mocker.patch("pydefect.cli.vasp.make_efnv_correction.Ewald")
ewald = mocker.Mock()
ewald.lattice_energy = 1e3
ewald.atomic_site_potential.return_value = 1e4
mock_ewald.return_value = ewald
efnvc = make_efnv_correction(charge=2,
calc_results=mock_defect,
perfect_calc_results=mock_perfect,
dielectric_tensor=np.eye(3))
unit_conversion = 180.95128169876497
assert efnvc.charge == 2
assert efnvc.point_charge_correction == -4e3 * unit_conversion
assert efnvc.defect_region_radius == 5.0
assert efnvc.sites == [
PotentialSite("He", 5 * np.sqrt(2), 10.0, 2e4 * unit_conversion),
PotentialSite("He", 5 * np.sqrt(2), 20.0, 2e4 * unit_conversion),
PotentialSite("He", 5 * np.sqrt(2), 30.0, 2e4 * unit_conversion),
PotentialSite("Li", 5.0, 40.0, None),
]
def ortho_conventional():
lattice = Lattice.orthorhombic(5, 6, 7)
coords = [
[0.0, 0.0, 0.0],
[0.5, 0.5, 0.0],
[0.5, 0.0, 0.5],
[0.0, 0.5, 0.5],
[0.0, 0.0, 0.5],
[0.0, 0.5, 0.0],
[0.5, 0.0, 0.0],
[0.5, 0.5, 0.5],
]
return IStructure(lattice=lattice,
species=["H"] * 4 + ["He"] * 4,
coords=coords)
def setUpClass(cls):
coords = [[0, 0, 0], [0.75, 0.5, 0.75]]
lattice = Lattice([[3.8401979337, 0.00, 0.00],
[1.9200989668, 3.3257101909, 0.00],
[0.00, -2.2171384943, 3.1355090603]])
cls.struct_si = IStructure(lattice, ["Si"] * 2, coords)
cls.ref_incar = Incar.from_file(
os.path.join(module_dir, "..", "test_files", "setup_test",
"INCAR"))
cls.ref_poscar = Poscar.from_file(
os.path.join(module_dir, "..", "test_files", "setup_test",
"POSCAR"))
cls.ref_potcar = Potcar.from_file(
os.path.join(module_dir, "..", "test_files", "setup_test",
"POTCAR"))
cls.ref_kpoints = Kpoints.from_file(
os.path.join(module_dir, "..", "test_files", "setup_test",
"KPOINTS"))
def make_defect_entry(name: str, charge: int, perfect_structure: IStructure,
defect_structure: IStructure):
analyzer = DefectStructureComparator(perfect_structure, defect_structure)
species = []
frac_coords = []
for d, p in enumerate(analyzer.p_to_d):
if p is None:
site = defect_structure[d]
else:
site = perfect_structure[p]
species.append(site.specie)
frac_coords.append(site.frac_coords)
initial_structure = IStructure(perfect_structure.lattice, species,
frac_coords)
symmetrizer = StructureSymmetrizer(initial_structure)
return DefectEntry(name=name,
charge=charge,
structure=initial_structure,
site_symmetry=symmetrizer.point_group,
defect_center=tuple(analyzer.defect_center_coord))
def rhombohedral():
lattice = Lattice.rhombohedral(a=1, alpha=45)
coords = [[0.0, 0.0, 0.0]]
return IStructure(lattice=lattice, species=["H"], coords=coords)
def tetra_close_to_cubic():
lattice = Lattice.tetragonal(1.001 * 10 / sqrt(2), 10)
coords = [[0.0, 0.0, 0.0]]
return IStructure(lattice=lattice, species=["H"], coords=coords)
def test_num_atom_diff():
s1 = IStructure(Lattice.cubic(1), ["H", "He"], [[0] * 3] * 2)
s2 = IStructure(Lattice.cubic(1), ["H", "Li"], [[0] * 3] * 2)
assert num_atom_differences(s1, s2) == {"He": 1, "Li": -1}
def c_centered_monoclinic():
lattice = Lattice.monoclinic(3, 4, 5, 100)
coords = [[0.0, 0.0, 0.0], [0.5, 0.5, 0.0]]
return IStructure(lattice=lattice, species=["H", "H"], coords=coords)
def simple_cubic():
lattice = Lattice.cubic(1.0)
coords = [[0.0, 0.0, 0.0]]
return IStructure(lattice=lattice, species=["H"], coords=coords)
def a_centered_orthorhombic():
lattice = Lattice([[1, 0, 0], [0, 2, 3], [0, -2, 3]])
coords = [[0.5, 0.8, 0.8], [0.0, 0.3, 0.0], [0.0, 0.0, 0.3]]
return IStructure(lattice=lattice, species=["H"] * 3, coords=coords)
def elongated_tetragonal():
lattice = Lattice.tetragonal(a=1, c=3 * sqrt(2))
coords = [[0.0, 0.0, 0.0]]
return IStructure(lattice=lattice, species=["H"], coords=coords)
def bcc():
lattice = Lattice.cubic(1.0)
coords = [[0.0, 0.0, 0.0], [0.5, 0.5, 0.5]]
return IStructure(lattice=lattice, species=["H"] * 2, coords=coords)
def simple_cubic_2x1x1():
lattice = Lattice.orthorhombic(2.0, 1.0, 1.0)
coords = [[0.0, 0.0, 0.0], [0.5, 0.0, 0.0]]
return IStructure(lattice=lattice, species=["H", "H"], coords=coords)
def setUp(self):
coords = [[0, 0, 0]]
lattice = Lattice([[1, 0, 0], [0, 2, 0], [0, 0, 3]])
self.struct = IStructure(lattice, ["H"], coords)
def monoclinic():
lattice = Lattice.monoclinic(3, 4, 5, 100)
coords = [[0.0, 0.0, 0.0]]
return IStructure(lattice=lattice, species=["H"], coords=coords)
# "H" at [0.0, 0.0, 0.0] is removed here.
from vise.tests.helpers.assertion import assert_msonable, assert_json_roundtrip
perf_coords = [[0.0, 0.0, 0.0], [0.5, 0.5, 0.0], [0.5, 0.0, 0.5],
[0.0, 0.5, 0.5], [0.0, 0.0, 0.5], [0.0, 0.5, 0.0],
[0.5, 0.0, 0.0], [0.5, 0.5, 0.5]]
coords = deepcopy(perf_coords)
coords.pop(0)
perturbed_coords = \
[[0.5, 0.5, 0.0], [0.5, 0.0, 0.5], [0.0, 0.5, 0.5],
[0.0, 0.0, 0.501], [0.0, 0.501, 0.0], [0.501, 0.0, 0.0], [0.5, 0.5, 0.5]]
perfect = IStructure(Lattice.cubic(10.0), ["H"] * 4 + ["He"] * 4, perf_coords)
defect_structure = IStructure(Lattice.cubic(10.0), ["H"] * 3 + ["He"] * 4,
coords)
perturbed_defect = IStructure(Lattice.cubic(10.0), ["H"] * 3 + ["He"] * 4,
perturbed_coords)
@pytest.fixture
def defect_entry():
perturbed_sites = (PerturbedSite("He", 5.0, (0.0, 0.0, 0.5),
(0.0, 0.0, 0.501), 0.1),
PerturbedSite("He", 5.0, (0.0, 0.5, 0.0),
(0.0, 0.501, 0.0), 0.1),
PerturbedSite("He", 5.0, (0.5, 0.0, 0.0),
(0.501, 0.0, 0.0), 0.1))
return DefectEntry(name="Va_O1",