diff --git a/tests/test-grammar-integration.cpp b/tests/test-grammar-integration.cpp index 51e59f05a..39434fcf4 100644 --- a/tests/test-grammar-integration.cpp +++ b/tests/test-grammar-integration.cpp @@ -52,8 +52,8 @@ static bool match_string(const std::string & input, llama_grammar* grammar) { return false; } -static void test_grammar(const std::string & grammar_str, const std::vector & passing_strings, const std::vector & failing_strings) { - fprintf(stderr, "⚪ Testing grammar: %s\n", grammar_str.c_str()); +static void test_grammar(const std::string & test_desc, const std::string & grammar_str, const std::vector & passing_strings, const std::vector & failing_strings) { + fprintf(stderr, "⚪ Testing %s. Grammar: %s\n", test_desc.c_str(), grammar_str.c_str()); fflush(stderr); auto grammar = build_grammar(grammar_str); @@ -61,7 +61,7 @@ static void test_grammar(const std::string & grammar_str, const std::vectorstacks; - fprintf(stderr, " Checking valid strings:\n"); + fprintf(stderr, " Valid strings:\n"); // Passing strings for (const auto & test_string : passing_strings) { @@ -82,7 +82,7 @@ static void test_grammar(const std::string & grammar_str, const std::vectorstacks = original_stacks; } - fprintf(stderr, " Checking invalid strings:\n"); + fprintf(stderr, " Invalid strings:\n"); // Failing strings for (const auto & test_string : failing_strings) { @@ -108,24 +108,33 @@ static void test_grammar(const std::string & grammar_str, const std::vector