From bd0a2e2a81ce55119a29e19d593ddb9c090fd195 Mon Sep 17 00:00:00 2001
From: dawoodkhan82 <dawoodkhan82@gmail.com>
Date: Thu, 20 May 2021 15:16:16 -0400
Subject: [PATCH] unittests for mix

---
 test/test_mix.py        | 25 +++++++++++++++++++++++++
 test/test_transforms.py | 22 ----------------------
 2 files changed, 25 insertions(+), 22 deletions(-)
 create mode 100644 test/test_mix.py
 delete mode 100644 test/test_transforms.py

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()