folding-studio/tests/commands/predict/conftest.py

from unittest import mock

import pytest
from folding_studio.client import Response


@pytest.fixture()
def mock_send_request():
    with mock.patch(
        "folding_studio.client.Client.send_request",
        return_value=Response(
            output_signed_url="url", confidence_data={"metric": "value"}
        ),
    ) as m:
        yield m


@pytest.fixture()
def mock_download_results():
    with mock.patch(
        "folding_studio.client.Response.download_results",
    ) as m:
        yield m