mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
import unittest
|
|
import numpy as np
|
|
import gradio as gr
|
|
import gradio.inputs
|
|
import gradio.outputs
|
|
|
|
|
|
class TestInterface(unittest.TestCase):
|
|
def test_input_output_mapping(self):
|
|
io = gr.Interface(inputs='SketCHPad', outputs='textBOX', fn=lambda x: x)
|
|
self.assertIsInstance(io.input_interfaces[0], gradio.inputs.Sketchpad)
|
|
self.assertIsInstance(io.output_interfaces[0], gradio.outputs.Textbox)
|
|
|
|
def test_input_interface_is_instance(self):
|
|
inp = gradio.inputs.Image()
|
|
io = gr.Interface(inputs=inp, outputs='textBOX', fn=lambda x: x)
|
|
self.assertEqual(io.input_interfaces[0], inp)
|
|
|
|
def test_output_interface_is_instance(self):
|
|
out = gradio.outputs.Label()
|
|
io = gr.Interface(inputs='SketCHPad', outputs=out, fn=lambda x: x)
|
|
self.assertEqual(io.output_interfaces[0], out)
|
|
|
|
def test_prediction(self):
|
|
def model(x):
|
|
return 2*x
|
|
io = gr.Interface(inputs='textbox', outputs='textBOX', fn=model)
|
|
self.assertEqual(io.predict[0](11), 22)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |