mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-15 02:11:15 +08:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
import io
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
import unittest.mock as mock
|
||
|
|
from gradio import tunneling
|
||
|
|
|
||
|
|
|
||
|
|
# class TestTunneling(unittest.TestCase):
|
||
|
|
# pass
|
||
|
|
# @mock.patch("pkg_resources.require")
|
||
|
|
# def test_should_fail_with_distribution_not_found(self, mock_require):
|
||
|
|
|
||
|
|
class TestVerbose(unittest.TestCase):
|
||
|
|
"""Unncessary tests but just including them for the sake of completion."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.message = "print test"
|
||
|
|
self.capturedOutput = io.StringIO() # Create StringIO object
|
||
|
|
sys.stdout = self.capturedOutput # and redirect stdout.
|
||
|
|
|
||
|
|
def test_verbose_debug_true(self):
|
||
|
|
tunneling.verbose(self.message, debug_mode=True)
|
||
|
|
self.assertEqual(self.capturedOutput.getvalue().strip(), self.message)
|
||
|
|
|
||
|
|
def test_verbose_debug_false(self):
|
||
|
|
tunneling.verbose(self.message, debug_mode=False)
|
||
|
|
self.assertEqual(self.capturedOutput.getvalue().strip(), '')
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
sys.stdout = sys.__stdout__
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|