def test_uniform_directions_isotropy(self):
n = 1e5
for dim in [1, 2]:
u_0 = directions.get_uniform_vectors(n, dim, self.rng)
u_net = np.mean(u_0, axis=0)
u_net_mag = np.sqrt(np.sum(np.square(u_net)))
self.assertTrue(u_net_mag < 1e-2)
def test_uniform_directions_magnitude(self):
n = 1e5
for dim in [1, 2]:
u_0 = directions.get_uniform_vectors(n, dim, self.rng)
u_mags = np.sum(np.square(u_0), axis=-1)
self.assertTrue(np.allclose(u_mags, 1.0))
