handler_test.py

from handler import EndpointHandler


class TestEndpointHandler:
    _handler: EndpointHandler = EndpointHandler()

    def test_endpoint_handler(self):
        outputs: list = self._handler(data={"inputs": "Who is John Doe?"})
        assert len(outputs) == 1