File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change 11"""Tests for the RAG Generator."""
22
3+ import math
34import os
45from unittest .mock import MagicMock , patch
56
@@ -320,4 +321,4 @@ def test_generate_answer_none_openai_response(self):
320321 def test_constants (self ):
321322 """Test class constants have expected values."""
322323 assert Generator .MAX_TOKENS == 2000
323- assert Generator .TEMPERATURE == 0.5
324+ assert math . isclose ( Generator .TEMPERATURE , 0.5 )
Original file line number Diff line number Diff line change 11"""Tests for question detector functionality."""
22
3+ import math
34import os
45from unittest .mock import MagicMock , patch
56
@@ -182,7 +183,7 @@ def test_mocked_initialization(self):
182183 def test_class_constants (self , detector ):
183184 """Test that class constants are properly defined."""
184185 assert detector .MAX_TOKENS == 10
185- assert detector .TEMPERATURE == 0.1
186+ assert math . isclose ( detector .TEMPERATURE , 0.1 )
186187 assert detector .CHAT_MODEL == "gpt-4o-mini"
187188
188189 def test_openai_client_initialization (self , detector ):
You can’t perform that action at this time.
0 commit comments