aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2025-12-20 07:04:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

⏪ fix: fix bring back the test description for func generation

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-26 12:58:59 +02:00
parent df7e6f41cb
commit a7d5cec0bd
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dev_gpt/constants.py
View File

@@ -53,7 +53,7 @@ BLACKLISTED_PACKAGES = [
'clearbit' # because of installation issues on latest version 'clearbit' # because of installation issues on latest version
] ]
UNNECESSARY_PACKAGES = [ UNNECESSARY_PACKAGES = [
'fastapi', 'uvicorn', 'starlette' # because the wrappers are used instead 'flask', 'fastapi', 'uvicorn', 'starlette' # because the wrappers are used instead
] ]
LANGUAGE_PACKAGES = [ LANGUAGE_PACKAGES = [

15
test/unit/test_packages.py
View File

@@ -18,3 +18,18 @@ def test_package_parsing():
['gpt_3_5_turbo'], ['gpt_3_5_turbo'],
]): ]):
assert set(parsed) == set(expected) 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 == [
[],
[],
[],
]