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

formatting

Browse Source
This commit is contained in:
Abubakar Abid 2022-03-23 15:50:10 -07:00
parent fa27aa5876
commit 63d18ac02b
5 changed files with 39 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gradio/interface.py
View File

@ -18,8 +18,8 @@ from markdown_it import MarkdownIt
from mdit_py_plugins.footnote import footnote_plugin
from gradio import interpretation, utils
from gradio.components import Component, get_component_instance, Markdown, Button
from gradio.blocks import BlockContext, Row, Column
from gradio.blocks import BlockContext, Column, Row
from gradio.components import Button, Component, Markdown, get_component_instance
from gradio.external import load_from_pipeline, load_interface # type: ignore
from gradio.flagging import CSVLogger, FlaggingCallback # type: ignore
from gradio.inputs import State as i_State # type: ignore

1
test/test_flagging.py
View File

@ -8,7 +8,6 @@ import huggingface_hub
import gradio as gr
from gradio import flagging
# class TestDefaultFlagging(unittest.TestCase):
# def test_default_flagging_callback(self):
# with tempfile.TemporaryDirectory() as tmpdirname:

6
test/test_inputs.py
View File

@ -194,7 +194,7 @@ class TestSlider(unittest.TestCase):
"default": 15,
"name": "slider",
"label": "Slide Your Input",
"css": {}
"css": {},
},
)
@ -530,7 +530,7 @@ class TestAudio(unittest.TestCase):
"source": "upload",
"name": "audio",
"label": "Upload Your Audio",
"css": {}
"css": {},
},
)
self.assertIsNone(audio_input.preprocess(None))
@ -589,7 +589,7 @@ class TestFile(unittest.TestCase):
"file_count": "single",
"name": "file",
"label": "Upload Your File",
"css": {}
"css": {},
},
)
self.assertIsNone(file_input.preprocess(None))

57
test/test_outputs.py
View File

@ -1,8 +1,8 @@
import json
import os
import tempfile
import unittest
import json
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
@ -16,10 +16,8 @@ class TestTextbox(unittest.TestCase):
def test_in_interface(self):
iface = gr.Interface(lambda x: x[-1], "textbox", gr.outputs.Textbox())
self.assertEqual(iface.process(["Hello"])[0], ["o"])
iface = gr.Interface(
lambda x: x / 2, "number", gr.outputs.Textbox()
)
self.assertEqual(iface.process([10])[0], ['5.0'])
iface = gr.Interface(lambda x: x / 2, "number", gr.outputs.Textbox())
self.assertEqual(iface.process([10])[0], ["5.0"])
class TestLabel(unittest.TestCase):
@ -185,7 +183,7 @@ class TestHighlightedText(unittest.TestCase):
"name": "highlightedtext",
"label": None,
"show_legend": False,
"css": {}
"css": {},
},
)
ht = {"pos": "Hello ", "neg": "World"}
@ -232,10 +230,8 @@ class TestAudio(unittest.TestCase):
)
)
self.assertEqual(
audio_output.get_template_context(), {"name": "audio",
"label": None,
"source": "upload",
"css": {}}
audio_output.get_template_context(),
{"name": "audio", "label": None, "source": "upload", "css": {}},
)
self.assertTrue(
audio_output.deserialize(gr.test_data.BASE64_AUDIO["data"]).endswith(".wav")
@ -363,7 +359,6 @@ class TestDataframe(unittest.TestCase):
"col_width": None,
"default": [[None, None, None], [None, None, None], [None, None, None]],
"name": "dataframe",
},
)
with self.assertRaises(ValueError):
@ -373,14 +368,22 @@ class TestDataframe(unittest.TestCase):
to_save = dataframe_output.save_flagged(
tmpdirname, "dataframe_output", output, None
)
self.assertEqual(to_save, json.dumps({
"headers": ["num", "prime"],
"data": [[2, True], [3, True], [4, False]]
}))
self.assertEqual(
to_save,
json.dumps(
{
"headers": ["num", "prime"],
"data": [[2, True], [3, True], [4, False]],
}
),
)
self.assertEqual(
dataframe_output.restore_flagged(tmpdirname, to_save, None),
{"headers": ["num", "prime"],
"data": [[2, True], [3, True], [4, False]]})
{
"headers": ["num", "prime"],
"data": [[2, True], [3, True], [4, False]],
},
)
def test_in_interface(self):
def check_odd(array):
@ -417,12 +420,18 @@ class TestCarousel(unittest.TestCase):
carousel_output.get_template_context(),
{
"components": [
{"name": "textbox", "label": None, "default": "", "lines": 1,
"css": {}, 'placeholder': None}
{
"name": "textbox",
"label": None,
"default": "",
"lines": 1,
"css": {},
"placeholder": None,
}
],
"name": "carousel",
"label": "Disease",
"css": {}
"css": {},
},
)
output = carousel_output.postprocess(["Hello World", "Bye World"])
@ -473,13 +482,7 @@ class TestTimeseries(unittest.TestCase):
timeseries_output = gr.outputs.Timeseries(label="Disease")
self.assertEqual(
timeseries_output.get_template_context(),
{
"x": None,
"y": None,
"name": "timeseries",
"label": "Disease",
"css": {}
},
{"x": None, "y": None, "name": "timeseries", "label": "Disease", "css": {}},
)
data = {"Name": ["Tom", "nick", "krish", "jack"], "Age": [20, 21, 19, 18]}
df = pd.DataFrame(data)

7
test/test_routes.py
View File

@ -13,7 +13,7 @@ os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
class TestRoutes(unittest.TestCase):
def setUp(self) -> None:
self.io = Interface(lambda x: x+x, "text", "text")
self.io = Interface(lambda x: x + x, "text", "text")
self.app, _, _ = self.io.launch(prevent_thread_lock=True)
self.client = TestClient(self.app)
@ -37,8 +37,9 @@ class TestRoutes(unittest.TestCase):
self.assertEqual(response.status_code, 200)
def test_predict_route(self):
response = self.client.post("/api/predict/",
json={"data": ["test"], "fn_index": 0})
response = self.client.post(
"/api/predict/", json={"data": ["test"], "fn_index": 0}
)
self.assertEqual(response.status_code, 200)
output = dict(response.json())
self.assertEqual(output["data"], ["testtest"])