More robust test assertions

DaveDeCaprio · Jan 29, 2024 · 1705194 · 1705194
1 parent 0b9e960
commit 1705194
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/tests/integration_tests/integrations/test_google.py b/tests/integration_tests/integrations/test_google.py
@@ -146,7 +146,7 @@ async def test_google_speech_browser_v1_restart_too_quick():
     )
     out = await array_sink(stream)
     # We are checking for any valid result here really.
-    assert out == (["Hello, we recording is that working now and see?"] or [])
+    assert out == ["Hello, we recording is that working now and see?"] or out == []
 
 
 # @pytest.mark.asyncio

diff --git a/tests/unit_tests/langchain_extensions/test_pydantic.py b/tests/unit_tests/langchain_extensions/test_pydantic.py
@@ -170,10 +170,6 @@ def test_pydantic_full_next_question():
         }
     )
     logger.debug(f"Response:\n{ret}")
-    assert ret["parsed_result"] == ActionItem(
-        name="Clothes pickup",
-        description="Pick up my clothes from the dry cleaner on the corner",
-        assignee="Jeff",
-        due_date=datetime.date(2024, 1, 12),
-    )
+    assert isinstance(ret["parsed_result"], ActionItem)
+    assert ret["parsed_result"].due_date == datetime.date(2024, 1, 12)
     assert "clothes" in ret["output"].lower()