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

fix tests

Browse Source
This commit is contained in:
Ali Abid 2021-12-30 10:45:49 +00:00
parent a20871cd7c
commit da7c6c3701
4 changed files with 25 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frontend/src/gradio.jsx
View File

@ -202,7 +202,7 @@ export class GradioInterface extends React.Component {
return;
}
let component_state = { input_data: [], output_data: [] };
if (this.state.flag_index !== undefined) {
if (typeof this.state.flag_index === "number") {
component_state["flag_index"] = this.state.flag_index;
} else {
for (let [i, input_component] of this.props.input_components.entries()) {

2
gradio/utils.py
View File

@ -210,7 +210,7 @@ def get_config_file(interface):
for i, example in enumerate(examples):
for j, (component, cell) in enumerate(zip(interface.input_components + interface.output_components, example)):
examples[i][j] = component.restore_flagged(
interface, interface.flagging_dir, cell, interface.encryption_key if interface.encrypt else None)
interface.flagging_dir, cell, interface.encryption_key if interface.encrypt else None)
config["examples"] = examples
config["examples_dir"] = interface.examples
else:

32
test/test_inputs.py
View File

@ -34,7 +34,7 @@ class TestTextbox(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = text_input.save_flagged(tmpdirname, "text_input", "Hello World!", None)
self.assertEqual(to_save, "Hello World!")
restored = text_input.restore_flagged(to_save, None)
restored = text_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "Hello World!")
with self.assertWarns(DeprecationWarning):
@ -78,7 +78,7 @@ class TestNumber(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = numeric_input.save_flagged(tmpdirname, "numeric_input", 3, None)
self.assertEqual(to_save, 3)
restored = numeric_input.restore_flagged(to_save, None)
restored = numeric_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, 3)
self.assertIsInstance(numeric_input.generate_sample(), float)
numeric_input.set_interpret_parameters(steps=3, delta=1, delta_type="absolute")
@ -107,7 +107,7 @@ class TestSlider(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = slider_input.save_flagged(tmpdirname, "slider_input", 3, None)
self.assertEqual(to_save, 3)
restored = slider_input.restore_flagged(to_save, None)
restored = slider_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, 3)
self.assertIsInstance(slider_input.generate_sample(), int)
@ -142,7 +142,7 @@ class TestCheckbox(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = bool_input.save_flagged(tmpdirname, "bool_input", True, None)
self.assertEqual(to_save, True)
restored = bool_input.restore_flagged(to_save, None)
restored = bool_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, True)
self.assertIsInstance(bool_input.generate_sample(), bool)
bool_input = gr.inputs.Checkbox(default=True, label="Check Your Input")
@ -173,7 +173,7 @@ class TestCheckboxGroup(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = checkboxes_input.save_flagged(tmpdirname, "checkboxes_input", ["a", "c"], None)
self.assertEqual(to_save, '["a", "c"]')
restored = checkboxes_input.restore_flagged(to_save, None)
restored = checkboxes_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, ["a", "c"])
self.assertIsInstance(checkboxes_input.generate_sample(), list)
checkboxes_input = gr.inputs.CheckboxGroup(choices=["a", "b", "c"], default=["a", "c"],
@ -210,7 +210,7 @@ class TestRadio(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = radio_input.save_flagged(tmpdirname, "radio_input", "a", None)
self.assertEqual(to_save, 'a')
restored = radio_input.restore_flagged(to_save, None)
restored = radio_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "a")
self.assertIsInstance(radio_input.generate_sample(), str)
radio_input = gr.inputs.Radio(choices=["a", "b", "c"], default="a",
@ -246,7 +246,7 @@ class TestDropdown(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = dropdown_input.save_flagged(tmpdirname, "dropdown_input", "a", None)
self.assertEqual(to_save, 'a')
restored = dropdown_input.restore_flagged(to_save, None)
restored = dropdown_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "a")
self.assertIsInstance(dropdown_input.generate_sample(), str)
dropdown_input = gr.inputs.Dropdown(choices=["a", "b", "c"], default="a",
@ -289,7 +289,7 @@ class TestImage(unittest.TestCase):
self.assertEqual("image_input/0.png", to_save)
to_save = image_input.save_flagged(tmpdirname, "image_input", img, None)
self.assertEqual("image_input/1.png", to_save)
restored = image_input.restore_flagged(to_save, None)
restored = image_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "image_input/1.png")
self.assertIsInstance(image_input.generate_sample(), str)
@ -364,7 +364,7 @@ class TestAudio(unittest.TestCase):
self.assertEqual("audio_input/0.wav", to_save)
to_save = audio_input.save_flagged(tmpdirname, "audio_input", x_wav, None)
self.assertEqual("audio_input/1.wav", to_save)
restored = audio_input.restore_flagged(to_save, None)
restored = audio_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "audio_input/1.wav")
self.assertIsInstance(audio_input.generate_sample(), dict)
@ -422,11 +422,11 @@ class TestFile(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = file_input.save_flagged(tmpdirname, "file_input", [x_file], None)
self.assertEqual("file_input/0.pdf", to_save)
self.assertEqual("file_input/0", to_save)
to_save = file_input.save_flagged(tmpdirname, "file_input", [x_file], None)
self.assertEqual("file_input/1.pdf", to_save)
restored = file_input.restore_flagged(to_save, None)
self.assertEqual(restored, "file_input/1.pdf")
self.assertEqual("file_input/1", to_save)
restored = file_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "file_input/1")
self.assertIsInstance(file_input.generate_sample(), dict)
file_input = gr.inputs.File(label="Upload Your File")
@ -463,7 +463,7 @@ class TestDataframe(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = dataframe_input.save_flagged(tmpdirname, "dataframe_input", x_data, None)
self.assertEqual(json.dumps(x_data), to_save)
restored = dataframe_input.restore_flagged(to_save, None)
restored = dataframe_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(x_data, restored)
self.assertIsInstance(dataframe_input.generate_sample(), list)
@ -509,7 +509,7 @@ class TestVideo(unittest.TestCase):
self.assertEqual("video_input/0.mp4", to_save)
to_save = video_input.save_flagged(tmpdirname, "video_input", x_video, None)
self.assertEqual("video_input/1.mp4", to_save)
restored = video_input.restore_flagged(to_save, None)
restored = video_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(restored, "video_input/1.mp4")
self.assertIsInstance(video_input.generate_sample(), dict)
@ -552,7 +552,7 @@ class TestTimeseries(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = timeseries_input.save_flagged(tmpdirname, "video_input", x_timeseries, None)
self.assertEqual(json.dumps(x_timeseries), to_save)
restored = timeseries_input.restore_flagged(to_save, None)
restored = timeseries_input.restore_flagged(tmpdirname, to_save, None)
self.assertEqual(x_timeseries, restored)
self.assertIsInstance(timeseries_input.generate_sample(), dict)

14
test/test_outputs.py
View File

@ -71,7 +71,7 @@ class TestLabel(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdir:
to_save = label_output.save_flagged(tmpdir, "label_output", label, None)
self.assertEqual(to_save, '{"3": 0.7, "1": 0.2}')
self.assertEqual(label_output.restore_flagged(to_save, None), {"3": 0.7, "1": 0.2})
self.assertEqual(label_output.restore_flagged(tmpdirname, to_save, None), {"3": 0.7, "1": 0.2})
with self.assertRaises(ValueError):
label_output = gr.outputs.Label(type="unknown")
label_output.deserialize([1, 2, 3])
@ -158,7 +158,7 @@ class TestKeyValues(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = kv_output.save_flagged(tmpdirname, "kv_output", kv_list, None)
self.assertEqual(to_save, '[["a", 1], ["b", 2]]')
self.assertEqual(kv_output.restore_flagged(to_save, None), [["a", 1], ["b", 2]])
self.assertEqual(kv_output.restore_flagged(tmpdirname, to_save, None), [["a", 1], ["b", 2]])
def test_in_interface(self):
def letter_distribution(word):
@ -187,7 +187,7 @@ class TestHighlightedText(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = ht_output.save_flagged(tmpdirname, "ht_output", ht, None)
self.assertEqual(to_save, '{"pos": "Hello ", "neg": "World"}')
self.assertEqual(ht_output.restore_flagged(to_save, None), {"pos": "Hello ", "neg": "World"})
self.assertEqual(ht_output.restore_flagged(tmpdirname, to_save, None), {"pos": "Hello ", "neg": "World"})
def test_in_interface(self):
def highlight_vowels(sentence):
@ -248,7 +248,7 @@ class TestJSON(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = js_output.save_flagged(tmpdirname, "js_output", js, None)
self.assertEqual(to_save, '{"pos": "Hello ", "neg": "World"}')
self.assertEqual(js_output.restore_flagged(to_save, None), {"pos": "Hello ", "neg": "World"})
self.assertEqual(js_output.restore_flagged(tmpdirname, to_save, None), {"pos": "Hello ", "neg": "World"})
def test_in_interface(self):
def get_avg_age_per_gender(data):
@ -292,7 +292,7 @@ class TestFile(unittest.TestCase):
iface = gr.Interface(write_file, "text", "file")
self.assertDictEqual(iface.process(["hello world"])[0][0], {
'name': 'test.txt', 'size': 11, 'data': 'aGVsbG8gd29ybGQ='
'name': 'test.txt', 'size': 11, 'data': 'data:text/plain;base64,aGVsbG8gd29ybGQ='
})
file_output = gr.outputs.File()
with tempfile.TemporaryDirectory() as tmpdirname:
@ -327,7 +327,7 @@ class TestDataframe(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname:
to_save = dataframe_output.save_flagged(tmpdirname, "dataframe_output", output, None)
self.assertEqual(to_save, '[[2, true], [3, true], [4, false]]')
self.assertEqual(dataframe_output.restore_flagged(to_save, None), [[2, True], [3, True], [4, False]])
self.assertEqual(dataframe_output.restore_flagged(tmpdirname, to_save, None), [[2, True], [3, True], [4, False]])
def test_in_interface(self):
def check_odd(array):
@ -404,7 +404,7 @@ class TestTimeseries(unittest.TestCase):
to_save = timeseries_output.save_flagged(tmpdirname, "timeseries_output", output, None)
self.assertEqual(to_save, '{"headers": ["Name", "Age"], "data": [["Tom", 20], ["nick", 21], ["krish", 19], '
'["jack", 18]]}')
self.assertEqual(timeseries_output.restore_flagged(to_save, None), {"headers": ["Name", "Age"],
self.assertEqual(timeseries_output.restore_flagged(tmpdirname, to_save, None), {"headers": ["Name", "Age"],
"data": [["Tom", 20], ["nick", 21],
["krish", 19], ["jack", 18]]})