diff --git a/setup.py b/setup.py index 1445bb808..d3807f2ce 100644 --- a/setup.py +++ b/setup.py @@ -29,6 +29,7 @@ "docker", "pymongo", "pymysql", + "bs4", "beautifulsoup4", "feedparser", ] @@ -52,7 +53,6 @@ # released requires minimal_requires = [ - "bs4", "docstring_parser", "loguru==0.6.0", "tiktoken", diff --git a/tests/dashscope_services_test.py b/tests/dashscope_services_test.py index aeff36ceb..198325e7f 100644 --- a/tests/dashscope_services_test.py +++ b/tests/dashscope_services_test.py @@ -286,7 +286,7 @@ def test_dashscope_text_to_audio_success( # Verify the file operations mock_os_makedirs.assert_called_once_with(saved_dir, exist_ok=True) - mock_open_func.assert_called_once_with(f"{saved_dir}/{text}.wav", "wb") + mock_open_func.assert_called_once() # Expected result expected_result = ServiceResponse( @@ -295,7 +295,13 @@ def test_dashscope_text_to_audio_success( ) self.assertEqual(results.status, expected_result.status) - self.assertEqual(results.content, expected_result.content) + self.assertIn( + results.content, + [ + {"audio_path": f"{saved_dir}/{text}.wav"}, + {"audio_path": f"{saved_dir}\\{text}.wav"}, # For Windows + ], + ) if __name__ == "__main__":