diff --git a/tests/test_llms.py b/tests/test_llms.py index fa84a6a..bf98469 100644 --- a/tests/test_llms.py +++ b/tests/test_llms.py @@ -90,7 +90,8 @@ def test_get_strutured_output_ollama(self): """ self.assertIsInstance(response, Character) - self.assertEqual('Harry James Potter', response.name) + assert response.name.startswith('Harry') + assert response.name.endswith('Potter') if __name__ == '__main__': unittest.main()