mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
31 lines
891 B
Python
31 lines
891 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()
|