def test_int_p_r_is_ip_r(self):
import scipy.integrate
D = 1e-12
t = 1e-5
r0 = 5e-8
r = 2.5e-8
gf = mod.GreensFunction3D(D, r0)
ip = gf.ip_r(0.0, t)
self.assertEqual(0.0, ip)
maxr = 1e-6
resolution = 20
for i in range(1, resolution):
r = i * maxr / resolution
ip = gf.ip_r(r, t)
result = scipy.integrate.quad(gf.p_r, 0.0, r,
args=(t, ))
np = result[0]
self.assertAlmostEqual(0.0, (np-ip)/ip)
def test_ip_r_infinity_is_one(self):
D = 1e-12
t = 1e-5
r0 = 5e-8
r = 2.5e-8
gf = mod.GreensFunction3D(D, r0)
ip = gf.ip_r(numpy.inf, t)
self.assertEqual(1.0, ip)
def test_drawR_zerot_is_r0(self):
D = 1e-12
r0 = 2e-8
gf = mod.GreensFunction3D(D, r0)
t = 0.0
r = gf.drawR(0.5, t)
self.assertEqual(r, r0)
r = gf.drawR(0.0, t)
self.assertEqual(r, r0)
r = gf.drawR(1.0, t)
self.assertEqual(r, r0)
def test_drawR(self):
D = 1e-12
r0 = 2e-8
gf = mod.GreensFunction3D(D, r0)
t = 1e-3
r = gf.drawR(0.5, t)
self.failIf(r < 0.0)
r = gf.drawR(0.0, t)
self.failIf(r < 0.0)
r = gf.drawR(1.0, t)
self.failIf(r < 0.0)
def test_drawR_smallt(self):
D = 1e-12
r0 = 2e-8
gf = mod.GreensFunction3D(D, r0)
t = 1e-4
while t > 1e-60:
r = gf.drawR(0.5, t)
self.failIf(r < 0.0)
r = gf.drawR(0.0, t)
self.failIf(r < 0.0)
r = gf.drawR(1.0, t)
self.failIf(r < 0.0)
r = gf.drawR(1e-2, t)
self.failIf(r < 0.0)
t *= 1e-3
def test_draw_theta(self):
D = 1e-12
r0 = 5e-8
t = 1e-4
gf = mod.GreensFunction3D(D, r0)
#r = gf.drawR(0.5, r0, t)
r = r0
theta = gf.drawTheta(0.5, r, t)
self.failIf(theta < 0.0 or theta > numpy.pi)
theta = gf.drawTheta(0.0, r, t)
self.failIf(theta < 0.0 or theta > numpy.pi)
# rnd=1.0 fails on x87. it's not a big problem, but
# need to look at it later.
theta = gf.drawTheta(0.999999, r, t)
self.failIf(theta < 0.0 or theta > numpy.pi)
def test_int_p_theta_is_p_r(self):
import scipy.integrate
D = 1e-12
t = 1e-5
r0 = 5e-8
r = 2.5e-8
gf = mod.GreensFunction3D(D, r0)
ip = gf.ip_theta(numpy.pi, r, t)
result = scipy.integrate.quad(gf.p_theta, 0.0, numpy.pi,
args=(r, t))
np = result[0]
pr = gf.p_r(r, t) / (2 * numpy.pi * r * r)
self.assertAlmostEqual(0.0, (pr-ip)/pr)
self.assertAlmostEqual(0.0, (pr-np)/pr)
def test_int_p_theta_is_ip_theta(self):
import scipy.integrate
D = 1e-12
t = 1e-3
r0 = 5e-8
r = 2.5e-8
gf = mod.GreensFunction3D(D, r0)
ip = gf.ip_theta(0.0, r, t)
self.assertEqual(0.0, ip)
resolution = 20
for i in range(1, resolution):
theta = i * numpy.pi / resolution
ip = gf.ip_theta(theta, r, t)
result = scipy.integrate.quad(gf.p_theta, 0.0, theta,
args=(r, t))
np = result[0]
self.assertAlmostEqual(0.0, (np-ip)/ip)
def test_instantiation(self):
D = 1e-12
r0 = 5e-8
gf = mod.GreensFunction3D(D, r0)
self.failIf(gf == None)