def test_dd_parallel_random(self):
arr = random_periodic_upsample(self.nx, 16, seed=3)
nsamples = 300
# idxs = [(0,0), (100,100), (70, 0), (256, 128)]
idxs = zip(np.random.randint(0, self.nx, nsamples), np.random.randint(0, self.ny, nsamples))
den_diff = dens_diff_parallel(den=arr, idxs=idxs, outer_len=self.nx, nr=self.nx/4)
import pylab as pl
pl.ion()
pl.figure()
pl.imshow(arr, interpolation='nearest', cmap='hot')
pl.figure()
pl.plot(den_diff[:,0], den_diff[:,1], 'ro-')
pl.errorbar(den_diff[:,0], den_diff[:,1], yerr=den_diff[:,2])
print
print den_diff
raw_input('enter to continue')
def test_dd_parallel_const(self):
den = self.zz
den.fill(0.1)
den_diff = dens_diff_parallel(den=den, idxs=[(0,0)], outer_len=1.0, nr=self.nx/4)
ok_(np.allclose(den_diff[:,1], 0.0))
def test_dd_parallel_delta(self):
den = self.zz
den[self.nx/2, self.ny/2] = 1.0
den_diff = dens_diff_parallel(den=den, idxs=[(4,4)], outer_len=1.0, nr=self.nx/4)
ok_(not np.allclose(den_diff[:,1], 0.0))
def test_dd_parallel_zeros(self):
den_diff = dens_diff_parallel(den=self.zz, idxs=[(0,0)], outer_len=1.0, nr=self.nx/4)
ok_(np.allclose(den_diff[:,1], 0.0))