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

examples tests pass

Browse Source
This commit is contained in:
Abubakar Abid 2022-03-25 11:14:28 -07:00
parent 6eeaccc11e
commit 22f332b1ce
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
test/test_process_examples.py
View File

@ -6,24 +6,24 @@ from gradio import Interface, process_examples
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
# class TestProcessExamples(unittest.TestCase):
# def test_process_example(self):
# io = Interface(lambda x: "Hello " + x, "text", "text", examples=[["World"]])
# prediction, _ = process_examples.process_example(io, 0)
# self.assertEquals(prediction[0], "Hello World")
class TestProcessExamples(unittest.TestCase):
def test_process_example(self):
io = Interface(lambda x: "Hello " + x, "text", "text", examples=[["World"]])
prediction, _ = process_examples.process_example(io, 0)
self.assertEquals(prediction[0], "Hello World")
# def test_caching(self):
# io = Interface(
# lambda x: "Hello " + x,
# "text",
# "text",
# examples=[["World"], ["Dunya"], ["Monde"]],
# )
# io.launch(prevent_thread_lock=True)
# process_examples.cache_interface_examples(io)
# prediction = process_examples.load_from_cache(io, 1)
# io.close()
# self.assertEquals(prediction[0], "Hello Dunya")
def test_caching(self):
io = Interface(
lambda x: "Hello " + x,
"text",
"text",
examples=[["World"], ["Dunya"], ["Monde"]],
)
io.launch(prevent_thread_lock=True)
process_examples.cache_interface_examples(io)
prediction = process_examples.load_from_cache(io, 1)
io.close()
self.assertEquals(prediction[0], "Hello Dunya")
if __name__ == "__main__":