From 00006bb244faf1b95913d4d5966f9726a99544d9 Mon Sep 17 00:00:00 2001 From: Motov Yurii <109919500+YuriiMotov@users.noreply.github.com> Date: Fri, 25 Jul 2025 15:12:54 +0200 Subject: [PATCH] Simplify tests --- tests/test_response_model_default_factory.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/tests/test_response_model_default_factory.py b/tests/test_response_model_default_factory.py index 857502d9b..13c1f442b 100644 --- a/tests/test_response_model_default_factory.py +++ b/tests/test_response_model_default_factory.py @@ -1,18 +1,13 @@ -import functools -import random - from fastapi import FastAPI from fastapi.testclient import TestClient from pydantic import BaseModel, Field app = FastAPI() -messages = ["操作成功.", "Successful operation."] - class ResponseModel(BaseModel): code: int = 200 - message: str = Field(default_factory=functools.partial(random.choice, messages)) + message: str = Field(default_factory=lambda: "Successful operation.") @app.get( @@ -40,7 +35,7 @@ def test_response_model_has_default_factory_return_dict(): assert response.status_code == 200, response.text assert response.json()["code"] == 200 - assert response.json()["message"] in messages + assert response.json()["message"] == "Successful operation." def test_response_model_has_default_factory_return_model(): @@ -49,4 +44,4 @@ def test_response_model_has_default_factory_return_model(): assert response.status_code == 200, response.text assert response.json()["code"] == 200 - assert response.json()["message"] in messages + assert response.json()["message"] == "Successful operation."