def test_ball_smoke():
dit.math.prng.seed(0)
x = module.ball(3)
x_ = np.array([0.21324626, 0.4465436, -0.65226253])
assert np.allclose(x, x_)
dit.math.prng.seed(0)
x = module.ball(3, prng=dit.math.prng)
assert np.allclose(x, x_)
def test_ball_smoke():
dit.math.prng.seed(0)
x = module.ball(3)
x_ = np.array([ 0.21324626, 0.4465436 , -0.65226253])
np.testing.assert_allclose(x, x_)
dit.math.prng.seed(0)
x = module.ball(3, prng=dit.math.prng)
np.testing.assert_allclose(x, x_)
def test_base_with_size():
# size 3, 1
dit.math.prng.seed(0)
x = module.ball(3, 1)
x_ = np.array([[0.21324626, 0.4465436, -0.65226253]])
assert np.allclose(x, x_)
# size 4, 4
dit.math.prng.seed(0)
x = module.ball(4, 4)
x_ = np.array([[0.69375635, -0.36303705, 0.35293677, -0.05622584],
[-0.06238751, 0.24817385, 0.08706278, 0.87899164],
[0.70592518, 0.1128636, 0.41171971, 0.30951042],
[0.72885111, -0.1000816, 0.15272267, -0.41665039]])
assert np.allclose(x, x_)
def test_base_with_size():
# size 3, 1
dit.math.prng.seed(0)
x = module.ball(3, 1)
x_ = np.array([[ 0.21324626, 0.4465436 , -0.65226253]])
np.testing.assert_allclose(x, x_)
# size 4, 4
dit.math.prng.seed(0)
x = module.ball(4, 4)
x_ = np.array([
[ 0.69375635, -0.36303705, 0.35293677, -0.05622584],
[-0.06238751, 0.24817385, 0.08706278, 0.87899164],
[ 0.70592518, 0.1128636 , 0.41171971, 0.30951042],
[ 0.72885111, -0.1000816 , 0.15272267, -0.41665039]
])
np.testing.assert_allclose(x, x_)
