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
|
# fmt: off
import numpy as np
from ase.dft.bandgap import bandgap
class DummyCalculator:
def __init__(self, e_skn):
self.e_skn = np.array(e_skn, dtype=float)
self.ns, self.nk, self.nb = self.e_skn.shape
def get_ibz_k_points(self):
k = np.zeros((self.nk, 3))
k[:, 0] += np.arange(self.nk)
return k
def get_fermi_level(self):
return 0.0
def get_eigenvalues(self, kpt, spin):
return self.e_skn[spin, kpt]
def get_number_of_spins(self):
return self.ns
def test_bandgap():
def test(e_skn):
c = DummyCalculator(e_skn)
if c.ns == 1:
result = [bandgap(c), bandgap(c, direct=True)]
return [(gap, k1, k2) for gap, (s1, k1, n1), (s2, k2, n2) in result]
result = [bandgap(c), bandgap(c, direct=True)]
for gap, (s1, k1, n1), (s2, k2, n2) in result:
if k1 is not None:
assert gap == e_skn[s2][k2][n2] - e_skn[s1][k1][n1]
return [(gap, (s1, k1), (s2, k2))
for gap, (s1, k1, n1), (s2, k2, n2) in result]
r = test([[[-1, 1]]])
assert r == [(2, 0, 0), (2, 0, 0)]
r = test([[[-1, 2], [-3, 1]]])
assert r == [(2, 0, 1), (3, 0, 0)]
r = test([[[-1, 2, 3], [-1, -1, 1]]])
assert r == [(0, None, None), (0, None, None)]
r = test([[[-1, 2, 3], [-1, -1, 1]], [[-1, 2, 2], [-3, 1, 1]]])
assert r == [(0, (None, None), (None, None)),
(0, (None, None), (None, None))]
r = test([[[-1, 5], [-2, 2]], [[-2, 4], [-4, 1]]])
assert r == [(2, (0, 0), (1, 1)), (3, (0, 1), (1, 1))]
r = test([[[-1, -1, -1, 2]], [[-1, 1, 1, 1]]])
assert r == [(2, (0, 0), (1, 0)), (2, (1, 0), (1, 0))]
|
