Skip to content

Commit fc20cad

Browse files
committed
fix unittests
1 parent 661a84b commit fc20cad

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

vise/tests/input_set/test_structure_kpoints_generator.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -113,15 +113,15 @@ def test_manual_kpts():
113113
task=Task.structure_opt,
114114
kpt_mode=KpointsMode.uniform)
115115
generator.generate_input()
116-
assert generator.kpoints.kpts[0] == [2, 2, 2]
117-
assert generator.kpoints.kpts_shift == [0.0, 0.5, 0.0]
116+
assert generator.kpoints.kpts[0] == (2, 2, 2)
117+
assert generator.kpoints.kpts_shift == (0.0, 0.5, 0.0)
118118
generator = StructureKpointsGenerator(structure,
119119
task=Task.structure_opt,
120120
kpt_mode=KpointsMode.uniform,
121121
gamma_centered=True,
122122
kpt_density=1.0)
123123
generator.generate_input()
124-
assert generator.kpoints.kpts_shift == [0.0, 0.0, 0.0]
124+
assert generator.kpoints.kpts_shift == (0.0, 0.0, 0.0)
125125

126126

127127
def test_only_even_num_kpts(sc_structure):
@@ -130,15 +130,15 @@ def test_only_even_num_kpts(sc_structure):
130130
kpt_density=2.)
131131
generator.generate_input()
132132
num_kpt_list = generator.kpoints.kpts[0]
133-
assert num_kpt_list == [13, 13, 13]
133+
assert num_kpt_list == (13, 13, 13)
134134

135135
generator = StructureKpointsGenerator(sc_structure,
136136
task=Task.structure_opt,
137137
kpt_density=2.,
138138
only_even_num_kpts=True)
139139
generator.generate_input()
140140
num_kpt_list = generator.kpoints.kpts[0]
141-
assert num_kpt_list == [14, 14, 14]
141+
assert num_kpt_list == (14, 14, 14)
142142

143143

144144
def test_kpt_factor(sc_structure):
@@ -148,7 +148,7 @@ def test_kpt_factor(sc_structure):
148148
num_kpt_factor=2)
149149
generator.generate_input()
150150
num_kpt_list = generator.kpoints.kpts[0]
151-
assert num_kpt_list == [26, 26, 26]
151+
assert num_kpt_list == (26, 26, 26)
152152

153153

154154
def test_cluster_opt_kpoints(sc_structure):
@@ -157,9 +157,9 @@ def test_cluster_opt_kpoints(sc_structure):
157157
kpt_density=2.)
158158
generator.generate_input()
159159
num_kpt_list = generator.kpoints.kpts[0]
160-
assert num_kpt_list == [1, 1, 1]
160+
assert num_kpt_list == (1, 1, 1)
161161
kpt_shift = generator.kpoints.kpts_shift
162-
assert kpt_shift == [0, 0, 0]
162+
assert kpt_shift == (0.0, 0.0, 0.0)
163163

164164

165165
def test_band_path(sc_structure):
@@ -171,7 +171,7 @@ def test_band_path(sc_structure):
171171
# so num_kpt_list = [12, 12, 12] and kpt_shift = [0.5, 0.5, 0.5] are set.
172172
generator.generate_input()
173173
assert generator.kpoints.num_kpts == 44
174-
assert generator.kpoints.kpts[0] == [0.125, 0.125, 0.125]
174+
assert generator.kpoints.kpts[0] == (0.125, 0.125, 0.125)
175175

176176

177177
def test_oi_ti_bravais():
@@ -206,7 +206,7 @@ def test_hexagonal():
206206
task=Task.structure_opt,
207207
kpt_density=5)
208208
generator.generate_input()
209-
assert generator.kpoints.kpts_shift == [0.0, 0.0, 0.5]
209+
assert generator.kpoints.kpts_shift == (0.0, 0.0, 0.5)
210210

211211

212212
def test_conventional_input(tmpdir):

0 commit comments

Comments
 (0)