Function test_encode_image_wrong_dim_input_array

tests/test_hashing.py:130–137

(hasher, mocker)

def test_encode_image_wrong_dim_input_array(hasher, mocker):
    image_arr_4d = np.random.random((3, 3, 2, 5))
    check_image_array_hash_mocker = mocker.patch(
        'imagededup.methods.hashing.check_image_array_hash'
    )
    with pytest.raises(ValueError):
        hasher.encode_image(image_array=image_arr_4d)
    check_image_array_hash_mocker.assert_called_once_with(image_arr_4d)


def test_encode_image_returns_none_image_pp_not_array(hasher, mocker):