From c6e82618baeb5ac56f0ef852d8525a90d47e6b6e Mon Sep 17 00:00:00 2001 From: Kate Date: Sat, 15 Nov 2025 16:43:14 -0800 Subject: [PATCH] Fix equality checks with floating point values --- backend/tests/apps/ai/agent/tools/rag/generator_test.py | 3 ++- backend/tests/apps/slack/common/question_detector_test.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/tests/apps/ai/agent/tools/rag/generator_test.py b/backend/tests/apps/ai/agent/tools/rag/generator_test.py index 8b820adb35..014b2c7f19 100644 --- a/backend/tests/apps/ai/agent/tools/rag/generator_test.py +++ b/backend/tests/apps/ai/agent/tools/rag/generator_test.py @@ -1,5 +1,6 @@ """Tests for the RAG Generator.""" +import math import os from unittest.mock import MagicMock, patch @@ -320,4 +321,4 @@ def test_generate_answer_none_openai_response(self): def test_constants(self): """Test class constants have expected values.""" assert Generator.MAX_TOKENS == 2000 - assert Generator.TEMPERATURE == 0.5 + assert math.isclose(Generator.TEMPERATURE, 0.5) diff --git a/backend/tests/apps/slack/common/question_detector_test.py b/backend/tests/apps/slack/common/question_detector_test.py index bfa4a8a485..9ecc1b91aa 100644 --- a/backend/tests/apps/slack/common/question_detector_test.py +++ b/backend/tests/apps/slack/common/question_detector_test.py @@ -1,5 +1,6 @@ """Tests for question detector functionality.""" +import math import os from unittest.mock import MagicMock, patch @@ -182,7 +183,7 @@ def test_mocked_initialization(self): def test_class_constants(self, detector): """Test that class constants are properly defined.""" assert detector.MAX_TOKENS == 10 - assert detector.TEMPERATURE == 0.1 + assert math.isclose(detector.TEMPERATURE, 0.1) assert detector.CHAT_MODEL == "gpt-4o-mini" def test_openai_client_initialization(self, detector):