mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-09 02:00:44 +08:00
72039be93a
* Add code * add changeset * Update fuzzy-mirrors-scream.md * Update fuzzy-mirrors-scream.md * Fix tests * Update .changeset/fuzzy-mirrors-scream.md Co-authored-by: Abubakar Abid <abubakar@huggingface.co> * Fix code --------- Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com> Co-authored-by: Abubakar Abid <abubakar@huggingface.co>
78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
import inspect
|
|
import pathlib
|
|
from contextlib import contextmanager
|
|
|
|
import pytest
|
|
from gradio_client import Client
|
|
|
|
import gradio as gr
|
|
import gradio.utils
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line(
|
|
"markers", "flaky: mark test as flaky. Failure will not cause te"
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def test_file_dir():
|
|
return pathlib.Path(pathlib.Path(__file__).parent, "test_files")
|
|
|
|
|
|
@pytest.fixture
|
|
def io_components():
|
|
all_subclasses = gradio.utils.core_gradio_components()
|
|
subclasses = []
|
|
for subclass in all_subclasses:
|
|
if subclass in [
|
|
gr.components.FormComponent,
|
|
gr.State,
|
|
gr.LoginButton,
|
|
gr.LogoutButton,
|
|
]:
|
|
continue
|
|
if subclass in gr.components.FormComponent.__subclasses__():
|
|
continue
|
|
|
|
if "value" in inspect.signature(subclass.__init__).parameters:
|
|
subclasses.append(subclass)
|
|
|
|
return subclasses
|
|
|
|
|
|
@pytest.fixture
|
|
def connect():
|
|
@contextmanager
|
|
def _connect(demo: gr.Blocks, serialize=True, **kwargs):
|
|
_, local_url, _ = demo.launch(prevent_thread_lock=True, **kwargs)
|
|
try:
|
|
client = Client(local_url, serialize=serialize)
|
|
yield client
|
|
finally:
|
|
client.close()
|
|
demo.close()
|
|
|
|
return _connect
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def gradio_temp_dir(monkeypatch, tmp_path):
|
|
"""tmp_path is unique to each test function.
|
|
It will be cleared automatically according to pytest docs: https://docs.pytest.org/en/6.2.x/reference.html#tmp-path
|
|
"""
|
|
monkeypatch.setenv("GRADIO_TEMP_DIR", str(tmp_path))
|
|
return tmp_path
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clear_static_files():
|
|
"""Clears all static files from the _StaticFiles class.
|
|
|
|
This is necessary because the tests should be independent of one another.
|
|
"""
|
|
yield
|
|
from gradio import data_classes
|
|
|
|
data_classes._StaticFiles.clear()
|