def run_test(arr, *args, **kwargs): for fn in self.array_transforms: arg = fn(arr) self.match( array_ops.imag(arg, *args, **kwargs), # np.imag may return a scalar so we convert to a np.ndarray. np.array(np.imag(arg, *args, **kwargs)))
def isreal(x): return array_ops.imag(x) == 0
def iscomplex(x): return array_ops.imag(x) != 0