mirror of
https://github.com/gradio-app/gradio.git
synced 2024-11-27 01:40:20 +08:00
21 lines
671 B
Python
21 lines
671 B
Python
|
from pathlib import Path
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
import gradio
|
||
|
|
from gradio.reload import run_in_reload_mode
|
||
|
|
|
||
|
|
|
||
|
|
@patch("gradio.reload.os.system")
|
||
|
|
@patch("gradio.reload.sys")
|
||
|
|
def test_run_in_reload_mode(mock_sys, mock_system_call):
|
||
|
|
|
||
|
|
mock_sys.argv = ["gradio", "demo/calculator/run.py"]
|
||
|
|
run_in_reload_mode()
|
||
|
|
reload_command = mock_system_call.call_args[0][0]
|
||
|
|
gradio_dir = Path(gradio.__file__).parent
|
||
|
|
demo_dir = Path("demo/calculator/run.py").resolve().parent
|
||
|
|
|
||
|
|
assert "uvicorn demo.calculator.run:demo.app" in reload_command
|
||
|
|
assert f'--reload-dir "{gradio_dir}"' in reload_command
|
||
|
|
assert f'--reload-dir "{demo_dir}"' in reload_command
|