@@ -1902,28 +1902,21 @@ public void testIntervalAsString() {
19021902    public  void  testCategorizeSingleGrouping () {
19031903        assumeTrue ("requires Categorize capability" , EsqlCapabilities .Cap .CATEGORIZE_V5 .isEnabled ());
19041904
1905-         query ("from test | STATS COUNT(*) BY CATEGORIZE(first_name)" );
1906-         query ("from test | STATS COUNT(*) BY cat = CATEGORIZE(first_name)" );
1905+         query ("FROM test | STATS COUNT(*) BY CATEGORIZE(first_name)" );
1906+         query ("FROM test | STATS COUNT(*) BY cat = CATEGORIZE(first_name)" );
1907+         query ("FROM test | STATS COUNT(*) BY CATEGORIZE(first_name), emp_no" );
1908+         query ("FROM test | STATS COUNT(*) BY a = CATEGORIZE(first_name), b = emp_no" );
19071909
19081910        assertEquals (
1909-             "1:31: cannot use CATEGORIZE grouping function [CATEGORIZE(first_name)] with multiple groupings" ,
1910-             error ("from test | STATS COUNT(*) BY CATEGORIZE(first_name), emp_no" )
1911-         );
1912-         assertEquals (
1913-             "1:39: cannot use CATEGORIZE grouping function [CATEGORIZE(first_name)] with multiple groupings" ,
1911+             "1:39: CATEGORIZE grouping function [CATEGORIZE(first_name)] can only be in the first grouping expression" ,
19141912            error ("FROM test | STATS COUNT(*) BY emp_no, CATEGORIZE(first_name)" )
19151913        );
19161914        assertEquals (
1917-             "1:35: cannot use CATEGORIZE grouping function [CATEGORIZE(first_name)] with multiple groupings" ,
1918-             error ("FROM test | STATS COUNT(*) BY a = CATEGORIZE(first_name), b = emp_no" )
1919-         );
1920-         assertEquals (
1921-             "1:31: cannot use CATEGORIZE grouping function [CATEGORIZE(first_name)] with multiple groupings\n " 
1922-                 + "line 1:55: cannot use CATEGORIZE grouping function [CATEGORIZE(last_name)] with multiple groupings" ,
1915+             "1:55: CATEGORIZE grouping function [CATEGORIZE(last_name)] can only be in the first grouping expression" ,
19231916            error ("FROM test | STATS COUNT(*) BY CATEGORIZE(first_name), CATEGORIZE(last_name)" )
19241917        );
19251918        assertEquals (
1926-             "1:31: cannot use  CATEGORIZE grouping function [CATEGORIZE(first_name)] with multiple groupings " ,
1919+             "1:55:  CATEGORIZE grouping function [CATEGORIZE(first_name)] can only be in the first grouping expression " ,
19271920            error ("FROM test | STATS COUNT(*) BY CATEGORIZE(first_name), CATEGORIZE(first_name)" )
19281921        );
19291922    }
0 commit comments