Skip to content

Commit c32f2ea

Browse files
committed
Fix test_array_ordering test
1 parent ed9e413 commit c32f2ea

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

python/cufinufft/tests/test_array_ordering.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,19 +7,17 @@
77
import utils
88

99

10-
def test_type1_ordering(to_gpu, to_cpu, dtype=np.float32, shape=(16, 16, 16), M=4096, tol=1e-3):
11-
complex_dtype = utils._complex_dtype(dtype)
12-
10+
def test_type1_ordering(to_gpu, to_cpu, dtype=np.complex64, shape=(16, 16, 16), M=4096, tol=1e-3):
1311
k, c = utils.type1_problem(dtype, shape, M)
1412

1513
k_gpu = to_gpu(k)
1614
c_gpu = to_gpu(c)
1715

18-
plan = Plan(1, shape, eps=tol, dtype=complex_dtype)
16+
plan = Plan(1, shape, eps=tol, dtype=dtype)
1917

2018
plan.setpts(*k_gpu)
2119

22-
out = np.empty(shape, dtype=complex_dtype, order="F")
20+
out = np.empty(shape, dtype=dtype, order="F")
2321

2422
out_gpu = to_gpu(out)
2523

0 commit comments

Comments
 (0)