From 5e2e63deb27f1d0195dfaf1c3ecb38937727a029 Mon Sep 17 00:00:00 2001 From: winstxnhdw Date: Wed, 28 Feb 2024 04:31:44 +0800 Subject: [PATCH] style: annotate return type --- tests/test_translate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_translate.py b/tests/test_translate.py index 06c6bab..f238c96 100644 --- a/tests/test_translate.py +++ b/tests/test_translate.py @@ -1,5 +1,7 @@ # pylint: disable=missing-function-docstring,redefined-outer-name +from typing import Generator + from fastapi.testclient import TestClient from pytest import fixture @@ -7,7 +9,7 @@ @fixture() -def client(): +def client() -> Generator[TestClient, None, None]: with TestClient(initialise()) as client: yield client