def test_power_operators():
a = 2 * nd.ones(shape=LARGE_X)
# sqrt
result = nd.sqrt(a)
assert result[-1] == 1.4142135
assert result.shape == a.shape
# rsqrt
result = nd.rsqrt(a)
assert result[-1] == 0.70710677
assert result.shape == a.shape
# cbrt
result = nd.cbrt(a)
assert result[-1] == 1.2599211
assert result.shape == a.shape
# rcbrt
result = nd.rcbrt(a)
assert result[-1] == 0.7937005
assert result.shape == a.shape
# square
result = nd.square(a)
assert result[-1] == 4
assert result.shape == a.shape
# reciprocal
result = nd.reciprocal(a)
assert result[-1] == 0.5
assert result.shape == a.shape
def rsqrt(x):
return nd.rsqrt(x)
