mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-03 01:50:59 +08:00
9787d0284a
* Use tempdir for all tests * Disable telemetry automaticall for client * Use configuration * Push js * Final comments * Undo change
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import inspect
|
|
import pathlib
|
|
from contextlib import contextmanager
|
|
|
|
import pytest
|
|
from gradio_client import Client
|
|
|
|
import gradio as gr
|
|
|
|
|
|
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():
|
|
classes_to_check = gr.components.IOComponent.__subclasses__()
|
|
subclasses = []
|
|
|
|
while classes_to_check:
|
|
subclass = classes_to_check.pop()
|
|
children = subclass.__subclasses__()
|
|
|
|
if children:
|
|
classes_to_check.extend(children)
|
|
if "value" in inspect.signature(subclass).parameters:
|
|
subclasses.append(subclass)
|
|
|
|
return subclasses
|
|
|
|
|
|
@pytest.fixture
|
|
def connect():
|
|
@contextmanager
|
|
def _connect(demo: gr.Blocks, serialize=True):
|
|
_, local_url, _ = demo.launch(prevent_thread_lock=True)
|
|
try:
|
|
yield Client(local_url, serialize=serialize)
|
|
finally:
|
|
# A more verbose version of .close()
|
|
# because we should set a timeout
|
|
# the tests that call .cancel() can get stuck
|
|
# waiting for the thread to join
|
|
if demo.enable_queue:
|
|
demo._queue.close()
|
|
demo.is_running = False
|
|
demo.server.should_exit = True
|
|
demo.server.thread.join(timeout=1)
|
|
|
|
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
|