def test_complex_fn_array(self):
assert_allclose(imag_func(np.array([2+2.j, 3+1.j]), self.freal), [4,2])
def test_real_fn_array(self):
assert_allclose(imag_func(np.array([2,3]), self.freal), [0,0])
def test_imag_fn_array(self):
assert_allclose(imag_func(np.array([2, 1.j]), self.fimag), [4, 0])
def test_imag_fn(self):
assert_allclose(imag_func(2, self.fimag), 4)
def test_complex_fn(self):
assert_allclose(imag_func(2+2.j, self.freal), 4)
def test_complex_fn_array(self):
assert_allclose(imag_func(np.array([2 + 2.j, 3 + 1.j]), self.freal),
[4, 2])
def test_real_fn(self):
assert_allclose(imag_func(2, self.freal), 0)
def test_imag_fn_array(self):
assert_allclose(imag_func(np.array([2, 1.j]), self.fimag), [4, 0])
def test_real_fn_array(self):
assert_allclose(imag_func(np.array([2, 3]), self.freal), [0, 0])
def test_complex_fn(self):
assert_allclose(imag_func(2 + 2.j, self.freal), 4)
def test_imag_fn(self):
assert_allclose(imag_func(2, self.fimag), 4)
def test_real_fn(self):
assert_allclose(imag_func(2, self.freal), 0)
