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