def test_from_parameters(self): n = Uniform(0, 1) r = Uniform(0, np.pi / 2) spheres = [Sphere(n=n, r=r, center=[i, i, i]) for i in range(3)] spheres = Spheres(spheres) parameters = spheres.parameters self.assertEqual(spheres.guess, spheres.from_parameters(parameters))
def test_from_parameters_keeps_attributes(self): spheres = [ Sphere(n=np.random.rand(), r=0.1, center=[i, i, i]) for i in range(3) ] spheres = Spheres(spheres, warn="TEST") spheres = spheres.from_parameters(spheres.parameters) self.assertEqual(spheres.warn, "TEST")