diff --git a/test/unit/test_packages.py b/test/unit/test_packages.py index b3987e8..7212f00 100644 --- a/test/unit/test_packages.py +++ b/test/unit/test_packages.py @@ -1,16 +1,15 @@ from dev_gpt.options.generate.generator import Generator def test_package_parsing(): - packages_json_string = '''\ - [ + packages_json = [ ["PyPDF2", "gpt_3_5_turbo"], ["pdfminer.six", "gpt_3_5_turbo"], ["tika", "gpt_3_5_turbo"], [], ["gpt_3_5_turbo"] - ]''' + ] - parsed_packages = Generator.process_packages_json_string(packages_json_string) + parsed_packages = Generator.process_packages_json_string(packages_json) for parsed, expected in zip(parsed_packages, [ ['pypdf2', 'gpt_3_5_turbo'], ['pdfminer.six', 'gpt_3_5_turbo'],