File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -131,7 +131,8 @@ def test_retrieve_city_state_laws_returns_joined_docs(mock_rag_class):
131131 """Test that RAG results are joined with newlines."""
132132 mock_rag_class .return_value .search .return_value = "Doc1 content\n Doc2 content"
133133
134- result = retrieve_city_state_laws .func ( # type: ignore[unresolved-attribute]
134+ _func = getattr (retrieve_city_state_laws , "func" )
135+ result = _func (
135136 query = "eviction notice" ,
136137 state = UsaState ("or" ),
137138 city = OregonCity ("portland" ),
@@ -146,7 +147,8 @@ def test_retrieve_city_state_laws_empty_results(mock_rag_class):
146147 """Test behavior when RAG returns no documents."""
147148 mock_rag_class .return_value .search .return_value = ""
148149
149- result = retrieve_city_state_laws .func ( # type: ignore[unresolved-attribute]
150+ _func = getattr (retrieve_city_state_laws , "func" )
151+ result = _func (
150152 query = "obscure law" ,
151153 state = UsaState ("or" ),
152154 runtime = MagicMock (),
@@ -174,6 +176,7 @@ def test_generate_letter_empty_string(mock_get_stream_writer):
174176 mock_writer = MagicMock ()
175177 mock_get_stream_writer .return_value = mock_writer
176178
177- result = generate_letter .func (letter = "" ) # type: ignore[unresolved-attribute]
179+ _func = getattr (generate_letter , "func" )
180+ result = _func (letter = "" )
178181 mock_writer .assert_called_once_with ({"type" : "letter" , "content" : "" })
179182 assert result == "Letter generated successfully."
You can’t perform that action at this time.
0 commit comments