gradio

1
import pytest
2

3
from gradio import Interface, networking
4

5

6
@pytest.mark.flaky
7
def test_setup_tunnel():
8
    io = Interface(lambda x: x, "number", "number")
9
    io.launch(show_error=True, prevent_thread_lock=True)
10
    share_url = networking.setup_tunnel(
11
        io.server_name, io.server_port, io.share_token, io.share_server_address
12
    )
13
    assert isinstance(share_url, str)
14

15

16
@pytest.mark.flaky
17
def test_setup_custom_tunnel():
18
    io = Interface(lambda x: x, "number", "number")
19
    io.launch(
20
        show_error=True,
21
        prevent_thread_lock=True,
22
        share_server_address="my-gpt-wrapper.com:7000",
23
    )
24
    share_url = networking.setup_tunnel(
25
        io.server_name, io.server_port, io.share_token, io.share_server_address
26
    )
27
    assert isinstance(share_url, str)
28