mirror of
https://github.com/aljazceru/dev-gpt.git
synced 2025-12-19 22:54:21 +01:00
35 lines
913 B
Python
35 lines
913 B
Python
from dev_gpt.options.generate.generator import Generator
|
|
|
|
def test_package_parsing():
|
|
packages_json_string = '''\
|
|
[
|
|
["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)
|
|
for parsed, expected in zip(parsed_packages, [
|
|
['pypdf2', 'gpt_3_5_turbo'],
|
|
['pdfminer.six', 'gpt_3_5_turbo'],
|
|
[],
|
|
['gpt_3_5_turbo'],
|
|
]):
|
|
assert set(parsed) == set(expected)
|
|
|
|
def test_package_parsing_blacklist():
|
|
packages_json_string = '''\
|
|
[
|
|
["Flask, flask_restful, flask_json, flask_cors, http.server"],
|
|
[],
|
|
[]
|
|
]'''
|
|
|
|
parsed_packages = Generator.process_packages_json_string(packages_json_string)
|
|
assert parsed_packages == [
|
|
[],
|
|
[],
|
|
[],
|
|
] |