Skip to content

Commit 56bfce3

Browse files
committed
Fix rpy_test to use current python API closes #5
1 parent 5a0311a commit 56bfce3

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

Python/rpy_test.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -83,29 +83,29 @@ def rotne_prager_tensor(r_vectors, eta, a):
8383

8484
# Setup FMM
8585
myFMM = PySTKFMM.Stk3DFMM(mult_order, max_pts, pbc, kernel)
86-
kdim, _, kdimTrg = myFMM.getKernelDimension(kernel)
86+
kdim, _, kdimTrg = myFMM.get_kernel_dimension(kernel)
8787

8888
# Create sources and target values
8989
src_value = np.random.randn(nsrc, kdim)
9090
src_value[:, 3] = a
9191
trg_value = np.zeros((ntrg, kdimTrg))
9292
if rank == 0:
93-
myFMM.showActiveKernels()
93+
myFMM.show_active_kernels()
9494
print('kdimSL = ', kdim)
9595
print('kdimTrg = ', kdimTrg)
9696

9797
# Set tree
98-
myFMM.setBox(np.array([0.0, 0.0, 0.0]), 2.0)
99-
myFMM.setPoints(nsrc, src_coord, ntrg, trg_coord, 0, 0)
100-
myFMM.setupTree(kernel)
98+
myFMM.set_box(np.array([0.0, 0.0, 0.0]), 2.0)
99+
myFMM.set_points(src_coord, trg_coord, np.empty(shape=(0,0)))
100+
myFMM.setup_tree(kernel)
101101

102102
# Evaluate FMM
103-
myFMM.evaluateFMM(kernel, nsrc, src_value, ntrg, trg_value, 0, 0)
103+
myFMM.evaluate_fmm(kernel, src_value, trg_value, np.empty(shape=(0,0)))
104104

105105
# Clear FMM and evaluate again
106106
trg_value[:,:] = 0
107-
myFMM.clearFMM(kernel)
108-
myFMM.evaluateFMM(kernel, nsrc, src_value, ntrg, trg_value, 0, 0)
107+
myFMM.clear_fmm(kernel)
108+
myFMM.evaluate_fmm(kernel, src_value, trg_value, np.empty(shape=(0,0)))
109109

110110
comm.Barrier()
111111

0 commit comments

Comments
 (0)