fix: fixed ctags test

ErikBjare · Oct 31, 2023 · fa7d409 · fa7d409
1 parent 8616718
commit fa7d409
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/gptme/tools/context.py b/gptme/tools/context.py
@@ -57,7 +57,7 @@ def ctags() -> str:
         try:
             tags.append(json.loads(line))
         except json.JSONDecodeError:
-            output += ["  failed to parse: {line}"]
+            output += [f"  failed to parse: {line}"]
             break
 
     files = {tag["path"] for tag in tags}

diff --git a/tests/test_tools_context.py b/tests/test_tools_context.py
@@ -24,4 +24,6 @@ def test_ctags(shell):
         pytest.skip("ctags not installed")
 
     output = ctags()
-    assert "function" in output, f"Expected 'def' in output: {output}"
+    expected_strings = ["def", "class", "gptme"]
+    for s in expected_strings:
+        assert s in output, f"Expected '{s}' in output: {output}"