def test_adding_newly_tied_scatterer(self):
n = Uniform(0, 1)
r = Uniform(0, 1)
spheres = [Sphere(n=n, r=r, center=[1, 1, 1])]
spheres = Spheres(spheres)
spheres.add(Sphere(n=n, r=r, center=[2, 2, 2]))
expected_keys = {
'n', 'r', '0:center.0', '0:center.1', '0:center.2', '1:center.0',
'1:center.1', '1:center.2'
}
self.assertEqual(set(spheres.parameters.keys()), expected_keys)
def test_adding_untied_scatterer(self):
n = Uniform(0, 1)
r = Uniform(0, 1)
spheres = [Sphere(n=n, r=r, center=[i, i, i]) for i in range(2)]
spheres = Spheres(spheres)
spheres.add(Sphere(n=1, r=1, center=[2, 2, 2]))
expected_keys = {
'n', 'r', '0:center.0', '0:center.1', '0:center.2', '1:center.0',
'1:center.1', '1:center.2', '2:n', '2:r', '2:center.0',
'2:center.1', '2:center.2'
}
self.assertEqual(set(spheres.parameters.keys()), expected_keys)
