diff --git a/dev_gpt/constants.py b/dev_gpt/constants.py index 77403ef..06f194b 100644 --- a/dev_gpt/constants.py +++ b/dev_gpt/constants.py @@ -53,7 +53,7 @@ BLACKLISTED_PACKAGES = [ 'clearbit' # because of installation issues on latest version ] UNNECESSARY_PACKAGES = [ - 'fastapi', 'uvicorn', 'starlette' # because the wrappers are used instead + 'flask', 'fastapi', 'uvicorn', 'starlette' # because the wrappers are used instead ] LANGUAGE_PACKAGES = [ diff --git a/test/unit/test_packages.py b/test/unit/test_packages.py index 73a3cd0..af6c4ca 100644 --- a/test/unit/test_packages.py +++ b/test/unit/test_packages.py @@ -17,4 +17,19 @@ def test_package_parsing(): [], ['gpt_3_5_turbo'], ]): - assert set(parsed) == set(expected) \ No newline at end of file + assert set(parsed) == set(expected) + +def test_package_parsing_blacklist(): + packages_json_string = '''\ + [ + ["Flask"], + [], + [] + ]''' + + parsed_packages = Generator.process_packages_json_string(packages_json_string) + assert parsed_packages == [ + [], + [], + [], + ] \ No newline at end of file