mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
import unittest
|
|
import gradio as gr
|
|
from gradio import mix
|
|
import os
|
|
|
|
|
|
"""
|
|
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.outputs.Textbox())
|
|
io2 = gr.Interface(lambda x: x + "!", "textbox", gr.outputs.Textbox())
|
|
series = mix.Series(io1, io2)
|
|
self.assertEqual(series.process(["Hello"])[0], ["Hello World!"])
|
|
|
|
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)
|
|
output = series("test/test_data/lion.jpg")
|
|
self.assertGreater(output['lion'], 0.5)
|
|
|
|
|
|
class TestParallel(unittest.TestCase):
|
|
def test_in_interface(self):
|
|
io1 = gr.Interface(lambda x: x + " World 1!", "textbox",
|
|
gr.outputs.Textbox())
|
|
io2 = gr.Interface(lambda x: x + " World 2!", "textbox",
|
|
gr.outputs.Textbox())
|
|
parallel = mix.Parallel(io1, io2)
|
|
self.assertEqual(parallel.process(["Hello"])[0], ["Hello World 1!",
|
|
"Hello World 2!"])
|
|
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)
|
|
hello_es, hello_de = parallel("Hello")
|
|
self.assertIn("hola", hello_es.lower())
|
|
self.assertIn("hallo", hello_de.lower())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|