Fixing examples

Author: mikaem

examples/darray.py

@@ -50,7 +50,7 @@
 z[:] = MPI.COMM_WORLD.Get_rank()
 g0 = z.get((0, slice(None), 0))
 z2 = z.redistribute(2)
-z = z2.redistribute(darray=z)
+z = z2.redistribute(out=z)
 g1 = z.get((0, slice(None), 0))
 assert np.all(g0 == g1)
 s0 = MPI.COMM_WORLD.reduce(np.linalg.norm(z)**2)
@@ -69,14 +69,14 @@
 if MPI.COMM_WORLD.Get_rank() == 0:
     assert abs(s0-s1) < 1e-12
 
-z1 = z0.redistribute(darray=z1)
-z0 = z1.redistribute(darray=z0)
+z1 = z0.redistribute(out=z1)
+z0 = z1.redistribute(out=z0)
 
 N = (6, 6, 6, 6, 6)
 m0 = DistArray(N, dtype=float, alignment=2)
 m0[:] = MPI.COMM_WORLD.Get_rank()
 m1 = m0.redistribute(4)
-m0 = m1.redistribute(darray=m0)
+m0 = m1.redistribute(out=m0)
 s0 = MPI.COMM_WORLD.reduce(np.linalg.norm(m0)**2)
 s1 = MPI.COMM_WORLD.reduce(np.linalg.norm(m1)**2)
 if MPI.COMM_WORLD.Get_rank() == 0:

examples/transforms.py

@@ -1,7 +1,7 @@
 import functools
 import numpy as np
 from mpi4py import MPI
-from mpi4py_fft import PFFT, DistArray
+from mpi4py_fft import PFFT, newDistArray
 from mpi4py_fft.fftw import dctn, idctn
 
 # Set global size of the computational box
@@ -17,16 +17,16 @@
 
 assert fft.axes == pfft.axes
 
-u = DistArray(pfft=fft, forward_output=False)
+u = newDistArray(fft, forward_output=False)
 u[:] = np.random.random(u.shape).astype(u.dtype)
 
-u_hat = DistArray(pfft=fft, forward_output=True)
+u_hat = newDistArray(fft, forward_output=True)
 u_hat = fft.forward(u, u_hat)
 uj = np.zeros_like(u)
 uj = fft.backward(u_hat, uj)
 assert np.allclose(uj, u)
 
-u_padded = DistArray(pfft=pfft, forward_output=False)
+u_padded = newDistArray(pfft, forward_output=False)
 uc = u_hat.copy()
 u_padded = pfft.backward(u_hat, u_padded)
 u_hat = pfft.forward(u_padded, u_hat)