Function test_complex_params

python/tests/kernel/test_kernel_complex.py:21–50

Test that we can pass complex lists to kernel functions.

()

def test_complex_params():
    """Test that we can pass complex lists to kernel functions."""

    # Pass a list of complex as a parameter
    c = [.70710678 + 1j, 2j, 0., 0.70710678]

    @cudaq.kernel
    def complex_vec_param(vec: list[complex], i: int) -> complex:
        return vec[i]

    for i in range(len(c)):
        is_close(c[i].real, complex_vec_param(c, i).real)
    for i in range(len(c)):
        is_close(c[i].imag, complex_vec_param(c, i).imag)

    @cudaq.kernel
    def complex_vec_param_real(vec: list[complex], i: int) -> float:
        v = vec[i]
        return v.real

    for i in range(len(c)):
        assert is_close(c[i].real, complex_vec_param_real(c, i))

    @cudaq.kernel
    def complex_vec_param_imag(vec: list[complex], i: int) -> float:
        v = vec[i]
        return v.imag

    for i in range(len(c)):
        assert is_close(c[i].imag, complex_vec_param_imag(c, i))


def test_complex_capture():