mirror of
https://github.com/gradio-app/gradio.git
synced 2025-02-11 11:19:58 +08:00
9918a9540b
* Mark tests as flaky and run in separate job * Fix typo * Fix typo * Add test type to config * Add quotes * Rename job to test * Rename job to test
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
import os
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
import gradio as gr
|
|
from gradio import mix
|
|
from gradio.external import TooManyRequestsError
|
|
|
|
"""
|
|
WARNING: Some of these tests have an external dependency: namely that Hugging Face's Hub and Space APIs do not change, and they keep their most famous models up. So if, e.g. Spaces is down, then these test will not pass.
|
|
"""
|
|
|
|
|
|
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
|
|
|
|
|
|
class TestSeries(unittest.TestCase):
|
|
def test_in_interface(self):
|
|
io1 = gr.Interface(lambda x: x + " World", "textbox", gr.Textbox())
|
|
io2 = gr.Interface(lambda x: x + "!", "textbox", gr.Textbox())
|
|
series = mix.Series(io1, io2)
|
|
self.assertEqual(series.process(["Hello"]), ["Hello World!"])
|
|
|
|
@pytest.mark.flaky
|
|
def test_with_external(self):
|
|
io1 = gr.Interface.load("spaces/abidlabs/image-identity")
|
|
io2 = gr.Interface.load("spaces/abidlabs/image-classifier")
|
|
series = mix.Series(io1, io2)
|
|
try:
|
|
output = series("gradio/test_data/lion.jpg")
|
|
self.assertGreater(output["lion"], 0.5)
|
|
except TooManyRequestsError:
|
|
pass
|
|
|
|
|
|
class TestParallel(unittest.TestCase):
|
|
def test_in_interface(self):
|
|
io1 = gr.Interface(lambda x: x + " World 1!", "textbox", gr.Textbox())
|
|
io2 = gr.Interface(lambda x: x + " World 2!", "textbox", gr.Textbox())
|
|
parallel = mix.Parallel(io1, io2)
|
|
self.assertEqual(
|
|
parallel.process(["Hello"]), ["Hello World 1!", "Hello World 2!"]
|
|
)
|
|
|
|
def test_multiple_return_in_interface(self):
|
|
io1 = gr.Interface(
|
|
lambda x: (x, x + x), "textbox", [gr.Textbox(), gr.Textbox()]
|
|
)
|
|
io2 = gr.Interface(lambda x: x + " World 2!", "textbox", gr.Textbox())
|
|
parallel = mix.Parallel(io1, io2)
|
|
self.assertEqual(
|
|
parallel.process(["Hello"]), ["Hello", "HelloHello", "Hello World 2!"]
|
|
)
|
|
|
|
@pytest.mark.flaky
|
|
def test_with_external(self):
|
|
io1 = gr.Interface.load("spaces/abidlabs/english_to_spanish")
|
|
io2 = gr.Interface.load("spaces/abidlabs/english2german")
|
|
parallel = mix.Parallel(io1, io2)
|
|
try:
|
|
hello_es, hello_de = parallel("Hello")
|
|
self.assertIn("hola", hello_es.lower())
|
|
self.assertIn("hallo", hello_de.lower())
|
|
except TooManyRequestsError:
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|