test_subtract_images.py

import pyclesperanto_prototype as cle


import numpy as np

input1 = np.asarray([[1, 2, 3]])
input2 = np.asarray([[4, 5, 7]])


def test_subtract_images():
    reference = np.asarray([[3, 3, 4]])
    output = cle.subtract_images(input2, input1)
    result = cle.pull(output)
    assert np.array_equal(result, reference)