mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
31 lines
925 B
Python
31 lines
925 B
Python
import os
|
|
import unittest
|
|
|
|
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")
|
|
|
|
# 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__":
|
|
unittest.main()
|