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))