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

fixing tests

Browse Source
This commit is contained in:
Abubakar Abid 2022-01-09 23:49:03 -06:00
parent eb3c082039
commit 3a79fe754a
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test/test_networking.py
View File

@ -116,6 +116,8 @@ class TestFlagging(unittest.TestCase):
callback = flagging.CSVLogger()
callback.flag = mock.MagicMock()
aiohttp.ClientSession.post = mock.MagicMock()
aiohttp.ClientSession.post.__aenter__ = None
aiohttp.ClientSession.post.__aexit__ = None
io = Interface(
lambda x: x, "text", "text",
analytics_enabled=True, flagging_callback=callback)
@ -130,18 +132,20 @@ class TestFlagging(unittest.TestCase):
io.close()
@mock.patch("aiohttp.ClientSession.post")
class TestInterpretation(unittest.TestCase):
def test_interpretation(self, mock_post):
def test_interpretation(self):
io = Interface(
lambda x: len(x), "text", "label",
interpretation="default", analytics_enabled=True)
app, _, _ = io.launch(prevent_thread_lock=True)
client = TestClient(app)
aiohttp.ClientSession.post = mock.MagicMock()
aiohttp.ClientSession.post.__aenter__ = None
aiohttp.ClientSession.post.__aexit__ = None
io.interpret = mock.MagicMock(return_value=(None, None))
response = client.post(
'/api/interpret/', json={"data": ["test test"]})
mock_post.assert_called()
aiohttp.ClientSession.post.assert_called()
self.assertEqual(response.status_code, 200)
io.close()