mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2024-12-21 02:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed broken tests

Browse Source
This commit is contained in:
Abubakar Abid 2020-07-08 17:02:17 -05:00
parent 78cdc465d9
commit 6d85108529
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gradio/outputs.py
View File

@ -55,7 +55,7 @@ class Label(AbstractOutput):
def postprocess(self, prediction):
if isinstance(prediction, str) or isinstance(prediction, Number):
return {"label": prediction}
return {"label": str(prediction)}
elif isinstance(prediction, dict):
sorted_pred = sorted(
prediction.items(),

6
test/test_interface.py
View File

@ -7,13 +7,13 @@ import gradio.outputs
class TestInterface(unittest.TestCase):
def test_input_output_mapping(self):
io = gr.Interface(inputs='SketCHPad', outputs='textBOX', fn=lambda x: x)
io = gr.Interface(inputs='SketCHPad', outputs='TexT', 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)
io = gr.Interface(inputs=inp, outputs='teXT', fn=lambda x: x)
self.assertEqual(io.input_interfaces[0], inp)
def test_output_interface_is_instance(self):
@ -24,7 +24,7 @@ class TestInterface(unittest.TestCase):
def test_prediction(self):
def model(x):
return 2*x
io = gr.Interface(inputs='textbox', outputs='textBOX', fn=model)
io = gr.Interface(inputs='textbox', outputs='TEXT', fn=model)
self.assertEqual(io.predict[0](11), 22)

2
test/test_networking.py
View File

@ -27,7 +27,7 @@ class TestGetAvailablePort(unittest.TestCase):
s.bind((networking.LOCALHOST_NAME, port)) # Bind to the port
new_port = networking.get_first_available_port(initial, final)
s.close()
self.assertFalse(port==new_port)
self.assertFalse(port == new_port)
# class TestSetSampleData(unittest.TestCase):