Skip to content

Commit cb942e2

Browse files
committed
Avoid np.linspace in fp kpoints tests.
1 parent 6525517 commit cb942e2

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

tests/fp/test_fp_kpoint.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,17 +18,16 @@ def kpt(line):
1818
return kpt
1919

2020

21+
VALS = [0., 0.25, 0.5, 0.75, 1.]
22+
2123
STRAIGHT_SIMPLE = [
22-
lambda t, s1=s1, s2=s2: [s1, s2, t] for s1 in np.linspace(0, 1, 5)
23-
for s2 in np.linspace(0, 1, 5)
24+
lambda t, s1=s1, s2=s2: [s1, s2, t] for s1 in VALS for s2 in VALS
2425
]
2526
STRAIGHT_SIMPLE.extend([
26-
lambda t, s1=s1, s2=s2: [s1, t, s2] for s1 in np.linspace(0, 1, 5)
27-
for s2 in np.linspace(0, 1, 5)
27+
lambda t, s1=s1, s2=s2: [s1, t, s2] for s1 in VALS for s2 in VALS
2828
])
2929
STRAIGHT_SIMPLE.extend([
30-
lambda t, s1=s1, s2=s2: [t, s1, s2] for s1 in np.linspace(0, 1, 5)
31-
for s2 in np.linspace(0, 1, 5)
30+
lambda t, s1=s1, s2=s2: [t, s1, s2] for s1 in VALS for s2 in VALS
3231
])
3332

3433
STRAIGHT_ANY_DIRECTION = [

0 commit comments

Comments
 (0)