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

2026-01-22 23:14:19 +01:00 · 2023-05-26 12:49:46 +02:00
parent 3e25d89c57
commit df7e6f41cb
6 changed files with 42 additions and 17 deletions
--- a/test/unit/test_packages.py
+++ b/test/unit/test_packages.py
@@ -0,0 +1,20 @@
+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)