Function test_context_manager

packages/python-sdk/tests/test_client.py:90–94

Test SimStudioClient as context manager.

(mock_close)

@patch('simstudio.requests.Session.close')
def test_context_manager(mock_close):
    """Test SimStudioClient as context manager."""
    with SimStudioClient(api_key="test-api-key") as client:
        assert client.api_key == "test-api-key"
    mock_close.assert_called_once()


@patch('simstudio.requests.Session.post')