Function test_complex_use

python/tests/kernel/test_kernel_complex.py:154–223

Test that we can use complex numbers inside kernel functions.

()

def test_complex_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.pi / 2 + 0j)
        return v.real

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

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

    t = np.sin(np.pi / 2 + 0j).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.pi / 2 + 0j)
        return v.real

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

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

    t = np.cos(np.pi / 2 + 0j).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.pi / 2 + 0j)
        return v.real

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

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

    t = np.sqrt(np.pi / 2 + 0j).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.pi / 2 + 0j)