Spaces:
Running
Running
File size: 4,150 Bytes
85b7206 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import os
import platform
import unittest
import json
## permit to import from parent directory also in
import sys
from pathlib import Path
parent = Path(__file__).parent.parent
sys.path.append(str(parent))
from tests import EVENTS_FOLDER, set_seed
from webApp import app
from constants import ALLOWED_ORIGIN, app_logger
class TestWebApp(unittest.TestCase):
def setUp(self):
self.app = app.test_client()
if platform.system() == "Windows" or platform.system() == "Win32":
os.environ["PYTHONUTF8"] = "1"
os.environ["IS_TESTING"] = "TRUE"
def tearDown(self):
if platform.system() == "Windows" or platform.system() == "Win32" and "PYTHONUTF8" in os.environ:
del os.environ["PYTHONUTF8"]
del os.environ["IS_TESTING"]
def test_main_route(self):
response = self.app.get('/')
self.assertEqual(response.status_code, 200)
self.assertIn(b'<html', response.data)
def test_getNext(self):
set_seed()
input_data = {"category": "1", "language": "de"}
response = self.app.post('/getSample', json=input_data, content_type='application/json')
self.assertEqual(response.status_code, 200)
output_data = response.data.decode("utf-8")
output_data = json.loads(output_data)
self.assertEqual(
output_data, {
'ipa_transcript': 'ziː vɪl niːmandɛːn haɪ̯raːtɛːn.',
'real_transcript': ['Sie will niemanden heiraten.'],
'transcript_translation': ''
})
def test_GetAccuracyFromRecordedAudio(self):
set_seed()
with open(EVENTS_FOLDER / "GetAccuracyFromRecordedAudio.json", "r") as src:
inputs_outputs = json.load(src)
inputs = inputs_outputs["inputs"]
inputs_by_language = inputs["en"]
loaded_body = inputs_by_language["body"]
input_data = json.loads(loaded_body)
response = self.app.post('/GetAccuracyFromRecordedAudio', json=input_data, content_type='application/json')
self.assertEqual(response.status_code, 200)
output_data = response.data.decode("utf-8")
output_data = json.loads(output_data)
for k in [
"real_transcript", "ipa_transcript", "pronunciation_accuracy", "real_transcripts", "matched_transcripts",
"pair_accuracy_category", "start_time", "end_time", "is_letter_correct_all_words"
]:
self.assertIn(k, output_data)
def test_GetAccuracyFromRecordedAudio_raises_exception(self):
set_seed()
with open(EVENTS_FOLDER / "GetAccuracyFromRecordedAudio.json", "r") as src:
inputs_outputs = json.load(src)
inputs = inputs_outputs["inputs"]
inputs_by_language = inputs["en"]
loaded_body = inputs_by_language["body"]
input_data = json.loads(loaded_body)
input_data["language"] = "wrong_language"
response = self.app.post('/GetAccuracyFromRecordedAudio', json=input_data, content_type='application/json')
self.assertEqual(response.status_code, 200)
output_data = response.data.decode("utf-8")
output_data = json.loads(output_data)
self.assertEqual(output_data, {})
def test_change_model(self):
from constants import MODEL_NAME_TESTING
set_seed()
input_data = {"modelName": MODEL_NAME_TESTING}
response = self.app.post('/changeModel', json=input_data, content_type='application/json')
self.assertEqual(response.status_code, 200)
output_data = response.data.decode("utf-8")
self.assertEqual(output_data, f'Model changed to {input_data["modelName"]}!')
def test_change_model_raises_exception(self):
set_seed()
input_data = {"modelName": "wrong_model"}
response = self.app.post('/changeModel', json=input_data, content_type='application/json')
self.assertEqual(response.status_code, 500)
output_data = response.data.decode("utf-8")
self.assertEqual(output_data, 'Internal server error')
if __name__ == '__main__':
unittest.main()
|