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)