test_combine_vertically.py

import pyclesperanto_prototype as cle
import numpy as np


def test_combine_vertically():
    test1 = cle.push(np.asarray([
        [1, 1],
        [1, 1]
    ]))
    test2 = cle.push(np.asarray([
        [2, 2],
        [2, 2]
    ]))

    reference = cle.push(np.asarray([
        [1, 1],
        [1, 1],
        [2, 2],
        [2, 2]
    ]))

    result = cle.combine_vertically(test1, test2)

    a = cle.pull(result)
    b = cle.pull(reference)

    print(a)
    print(b)

    assert (np.allclose(a, b, 0.01))