Tests added

kuznetsov_av/api.py

@@ -1,7 +1,9 @@
+import base64
+
+import kuznetsov_av.text_to_speech_converter as t2s
 from fastapi import FastAPI
 from pydantic import BaseModel
-import kuznetsov_av.text_to_speech_converter as t2s
-import base64
+
 
 class Request(BaseModel):
     """
@@ -21,11 +23,11 @@ class Response(BaseModel):
 app = FastAPI()
 
 @app.get('/')
-async def root() -> str:
+async def root() -> dict:
     """
     Root method of API.
     """
-    return '{"message": "Converter method: /text-to-speech/convert/"}'
+    return {"message": "Converter method: /text-to-speech/convert/"}
 
 @app.post('/text-to-speech/convert/')
 async def text_to_speech(entity: Request) -> Response:

kuznetsov_av/test_api.py

@@ -0,0 +1,23 @@
+from fastapi.testclient import TestClient
+from kuznetsov_av.api import Request, Response, app
+
+client = TestClient(app)
+
+def test_root():
+    response = client.get('/')
+
+    assert response.status_code == 200
+    assert response.json().get('message') is not None
+    assert response.json().get('message') == "Converter method: /text-to-speech/convert/"
+
+def test_text_to_speech():
+    response = client.post(
+        url='/text-to-speech/convert/',
+        json=Request(text='Test').model_dump()
+    )
+
+    assert response.status_code == 200
+    assert response.json().get('audio') is not None
+    assert type(response.json().get('audio')) == str
+    assert response.json().get('sampling_rate') is not None
+    assert type(response.json().get('sampling_rate')) == int

kuznetsov_av/text_to_speech_converter.py

@@ -1,9 +1,9 @@
-from transformers import pipeline
-import transformers.pipelines.text_to_audio
-from datasets import load_dataset
 import datasets.arrow_dataset
-import torch
 import numpy as np
+import torch
+import transformers.pipelines.text_to_audio
+from datasets import load_dataset
+from transformers import pipeline
 
 
 def load_model() -> transformers.pipelines.text_to_audio.TextToAudioPipeline: