test_endpoint_handler.py

import pytest

from handler import EndpointHandler


@pytest.mark.parametrize('payload', [
    {
        'inputs': {
            'query': 'I like apples',
            'documents': [
                'But I only eat the red ones',
                'Asia is a continent',
                'You can add them to smoothies'
            ]
        }
    }
])
def test_sanity(payload):
    assert EndpointHandler(path=".")(payload)