def test_dirichlet_bvp_spherical():
r0, r1 = x0, x1
r2 = (r0 + r1) / 2
no_condition = NoCondition()
# B.C. for the interior boundary (r_min)
net_f = FCNN(2, 1)
f = lambda th, ph: no_condition.enforce(net_f, th, ph)
# B.C. for the exterior boundary (r_max)
net_g = FCNN(2, 1)
g = lambda th, ph: no_condition.enforce(net_g, th, ph)
condition = DirichletBVPSpherical(r_0=r0, f=f, r_1=r1, g=g)
net = FCNN(3, 1)
theta = torch.rand(N_SAMPLES, 1) * np.pi
phi = torch.rand(N_SAMPLES, 1) * 2 * np.pi
r = r0 * ones
assert all_close(condition.enforce(net, r, theta, phi),
f(theta, phi)), "inner Dirichlet BC not satisfied"
r = r1 * ones
assert all_close(condition.enforce(net, r, theta, phi),
g(theta, phi)), "inner Dirichlet BC not satisfied"
condition = DirichletBVPSpherical(r_0=r2, f=f)
r = r2 * ones
assert all_close(condition.enforce(net, r, theta, phi),
f(theta, phi)), "single ended BC not satisfied"
def test_inf_dirichlet_bvp_spherical():
r0 = random.random()
r1 = 1e15
no_condition = NoCondition()
net_f, net_g = FCNN(2, 1), FCNN(2, 1)
# B.C. for the interior boundary (r=r_min)
f = lambda th, ph: no_condition.enforce(net_f, th, ph)
# B.C. for the exterior boundary (r=infinity)
g = lambda th, ph: no_condition.enforce(net_g, th, ph)
net = FCNN(3, 1)
condition = InfDirichletBVPSpherical(r_0=r0, f=f, g=g, order=1)
theta = torch.rand(10, 1) * np.pi
phi = torch.rand(10, 1) * (2 * np.pi)
r = r0 * ones
assert all_close(condition.enforce(net, r, theta, phi), f(theta, phi)), "inner DirichletBC not satisfied"
r = r1 * ones
assert all_close(condition.enforce(net, r, theta, phi), g(theta, phi)), "Infinity DirichletBC not satisfied"
def test_no_condition():
N_INPUTS = 5
N_OUTPUTS = 5
for n_in, n_out in zip(range(1, N_INPUTS), range(1, N_OUTPUTS)):
xs = [torch.rand(N_SAMPLES, 1, requires_grad=True) for _ in range(n_in)]
net = FCNN(n_in, n_out)
cond = NoCondition()
y_cond = cond.enforce(net, *xs)
y_raw = net(torch.cat(xs, dim=1))
assert (y_cond == y_raw).all()
def U(x):
cond = NoCondition()
nets = [FCNN(3, 1) for _ in range(3)]
return tuple(cond.enforce(net, *x) for net in nets)
def u(x):
cond = NoCondition()
net = FCNN(3, 1)
return cond.enforce(net, *x)
def scalar_field(x):
cond = NoCondition()
return cond.enforce(FCNN(3, 1), *x)
def vector_field(x):
cond = NoCondition()
return tuple(cond.enforce(FCNN(3, 1), *x) for _ in range(3))
