def test_abs2():
x = np.array([pi(np.float128) + 1j, pi(np.float128)*1j, 3])
def func(d):
x_ = np.array(x if d.kind == 'c' else x.real, dtype=d)
actual = abs2(x_)
expected = np.abs(x_**2)
assert_same(actual, expected)
abs2(x_, actual)
assert_same(actual, expected)
for dtype in DTYPES:
yield func, dtype
def direct(self, input, output):
func = pixlib._ang2pix_nest if self.nest else pixlib._ang2pix_ring
if self.convention.startswith('azimuth'):
i1, i2 = input[..., 1], input[..., 0]
else:
i1, i2 = input[..., 0], input[..., 1]
if 'elevation' in self.convention:
i1 = 0.5 * pi(self.dtype) - i1
func(self.nside, i1, i2, output)
def direct(self, input, output):
func = pixlib._ang2pix_nest if self.nest else pixlib._ang2pix_ring
if self.convention.startswith('azimuth'):
i1, i2 = input[..., 1], input[..., 0]
else:
i1, i2 = input[..., 0], input[..., 1]
if 'elevation' in self.convention:
i1 = 0.5 * pi(self.dtype) - i1
func(self.nside, i1, i2, output)
def direct(self, input, output):
input = input.astype(int)
func = pixlib._pix2ang_nest if self.nest else pixlib._pix2ang_ring
if self.convention.startswith('azimuth'):
o1, o2 = output[..., 1], output[..., 0]
else:
o1, o2 = output[..., 0], output[..., 1]
func(self.nside, input, o1, o2)
if 'elevation' in self.convention:
np.subtract(0.5 * pi(self.dtype), o1, o1)
def direct(self, input, output):
input = input.astype(int)
func = pixlib._pix2ang_nest if self.nest else pixlib._pix2ang_ring
if self.convention.startswith('azimuth'):
o1, o2 = output[..., 1], output[..., 0]
else:
o1, o2 = output[..., 0], output[..., 1]
func(self.nside, input, o1, o2)
if 'elevation' in self.convention:
np.subtract(0.5 * pi(self.dtype), o1, o1)
