diff --git a/test/test_mix.py b/test/test_mix.py new file mode 100644 index 0000000000..a964e5f2a9 --- /dev/null +++ b/test/test_mix.py @@ -0,0 +1,25 @@ +import unittest +import gradio as gr +from gradio import mix + +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!"]) + +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!"]) + + +if __name__ == '__main__': + unittest.main() diff --git a/test/test_transforms.py b/test/test_transforms.py deleted file mode 100644 index afe599aa3e..0000000000 --- a/test/test_transforms.py +++ /dev/null @@ -1,22 +0,0 @@ -import unittest -import gradio as gr -import PIL -import numpy as np -import scipy -import os - - -class TestParallel(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 = gr.transforms.Series(io1, io2) - self.assertEqual(series.process(["Hello"])[0], ["Hello world!"]) - # io2 = gr.Interface(lambda x: x / 2, "number", - # gr.outputs.Textbox(type="number")) - # self.assertEqual(iface.process([10])[0], [5]) - - -if __name__ == '__main__': - unittest.main()