mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-02-23 11:39:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add type hint to provide autocompletion for Request class instances. (#1642)

Browse Source
This commit is contained in:
Halil Ibrahim Bestil 2022-06-26 17:10:02 +03:00 committed by GitHub
parent 0f44cf8012
commit 3026b8f597
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gradio/utils.py
View File

@ -13,7 +13,7 @@ import warnings
from copy import deepcopy
from distutils.version import StrictVersion
from enum import Enum
from typing import TYPE_CHECKING, Any, Callable, Dict, List, NewType, Type
from typing import TYPE_CHECKING, Any, Callable, Dict, Generator, List, NewType, Type
import aiohttp
import analytics
@ -424,7 +424,7 @@ class Request:
# Create request
self._request = self._create_request(method, url, **kwargs)
def __await__(self):
def __await__(self) -> Generator[None, Any, "Request"]:
"""
Wrap Request's __await__ magic function to create request calls which are executed in one line.
"""

14
test/test_utils.py
View File

@ -200,7 +200,7 @@ async def client():
class TestRequest:
@pytest.mark.asyncio
async def test_get(self):
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.GET,
url="http://headers.jsontest.com/",
)
@ -210,7 +210,7 @@ class TestRequest:
@pytest.mark.asyncio
async def test_post(self):
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.POST,
url="https://reqres.in/api/users",
json={"name": "morpheus", "job": "leader"},
@ -228,7 +228,7 @@ class TestRequest:
id: str
createdAt: str
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.POST,
url="https://reqres.in/api/users",
json={"name": "morpheus", "job": "leader"},
@ -242,7 +242,7 @@ class TestRequest:
name: Literal[str] = "John"
job: str
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.POST,
url="https://reqres.in/api/users",
json={"name": "morpheus", "job": "leader"},
@ -261,7 +261,7 @@ class TestRequest:
validate_response_data.side_effect = Exception()
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.GET,
url="https://reqres.in/api/users",
exception_type=ResponseValidationException,
@ -275,7 +275,7 @@ class TestRequest:
return response
raise Exception
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.POST,
url="https://reqres.in/api/users",
json={"name": "morpheus", "job": "leader"},
@ -294,7 +294,7 @@ class TestRequest:
return response
raise Exception
client_response: Request = await Request(
client_response = await Request(
method=Request.Method.POST,
url="https://reqres.in/api/users",
json={"name": "morpheus", "job": "leader"},