Function test_np_complex128_use

python/tests/kernel/test_kernel_complex.py:332–401

Test that we can use complex numbers inside kernel functions.

()

def test_np_complex128_use():
    """Test that we can use complex numbers inside kernel functions."""

    # Use a complex inside np in a kernel (sin)
    @cudaq.kernel
    def complex_np_use_real() -> float:
        v = np.sin(np.complex128(np.pi / 2 + 1 + 1j))
        return v.real

    t = np.sin(np.complex128(np.pi / 2 + 1 + 1j)).real
    assert is_close(t, complex_np_use_real())

    @cudaq.kernel
    def complex_np_use_imag() -> float:
        v = np.sin(np.complex128(np.pi / 2 + 1 + 1j))
        return v.imag

    t = np.sin(np.complex128(np.pi / 2 + 1 + 1j)).imag
    assert is_close(t, complex_np_use_imag())

    # Use a complex inside np in a kernel (cos)
    @cudaq.kernel
    def complex_np_use_real() -> float:
        v = np.cos(np.complex128(np.pi / 2 + 1 + 1j))
        return v.real

    t = np.cos(np.complex128(np.pi / 2 + 1 + 1j)).real
    assert is_close(t, complex_np_use_real())

    @cudaq.kernel
    def complex_np_use_imag() -> float:
        v = np.cos(np.complex128(np.pi / 2 + 1 + 1j))
        return v.imag

    t = np.cos(np.complex128(np.pi / 2 + 1 + 1j)).imag
    assert is_close(t, complex_np_use_imag())

    # Use a complex inside np in a kernel (sqrt)
    @cudaq.kernel
    def complex_np_use_real() -> float:
        v = np.sqrt(np.complex128(np.pi / 2 + 1 + 1j))
        return v.real

    t = np.sqrt(np.complex128(np.pi / 2 + 1 + 1j)).real
    assert is_close(t, complex_np_use_real())

    @cudaq.kernel
    def complex_np_use_imag() -> float:
        v = np.sqrt(np.complex128(np.pi / 2 + 1 + 1j))
        return v.imag

    t = np.sqrt(np.complex128(np.pi / 2 + 1 + 1j)).imag
    assert is_close(t, complex_np_use_imag())

    # Use a complex inside np in a kernel (exp)
    @cudaq.kernel
    def complex_np_use_real() -> float:
        v = np.exp(np.complex128(np.pi / 2 + 1 + 1j))