Add another test

promptware · Nov 8, 2024 · fc523d9 · fc523d9
1 parent a204204
commit fc523d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/python/metaprompt/parse_metaprompt.py b/python/metaprompt/parse_metaprompt.py
@@ -119,7 +119,7 @@ def extract_tokens(input_text):
     input_stream = InputStream(input_text)
     lexer = MetaPromptLexer(input_stream)
     token_stream = CommonTokenStream(lexer)
-    token_stream.fill()  # Fetches all tokens at once
+    token_stream.fill()
     tokens = []
     for token in token_stream.tokens:
         token_name = (
@@ -138,7 +138,6 @@ def extract_tokens(input_text):
 
 
 def parse_metaprompt(prompt):
-    print(extract_tokens(prompt))
     stream = InputStream(prompt)
     lexer = MetaPromptLexer(stream)
     stream = CommonTokenStream(lexer)

diff --git a/python/metaprompt/tests/test_parser.py b/python/metaprompt/tests/test_parser.py
@@ -95,6 +95,6 @@ def test_dummy_meta3():
     assert result["exprs"] == [t("["), t("a")]
 
 
-# def test_dummy_meta2():
-#     result = parse_metaprompt("[[][]]")
-#     assert result["exprs"] == [t("[[][]]")]
+def test_dummy_meta2():
+    result = parse_metaprompt("[[][]]")
+    assert result["exprs"] == [t("["), t("["), t("]"), t("["), t("]"), t("]")]