Update NCIdlFunctionsToken.scala
diff --git a/nlpcraft/src/test/scala/org/apache/nlpcraft/model/intent/idl/compiler/functions/NCIdlFunctionsToken.scala b/nlpcraft/src/test/scala/org/apache/nlpcraft/model/intent/idl/compiler/functions/NCIdlFunctionsToken.scala
index 58363a4..7ddaa74 100644
--- a/nlpcraft/src/test/scala/org/apache/nlpcraft/model/intent/idl/compiler/functions/NCIdlFunctionsToken.scala
+++ b/nlpcraft/src/test/scala/org/apache/nlpcraft/model/intent/idl/compiler/functions/NCIdlFunctionsToken.scala
@@ -195,6 +195,16 @@
                 idlCtx = mkIdlContext(Seq(tok1, tok2, tok3))
             ),
             TestDesc(
+                truth = "size(tok_all()) == 3",
+                token = tok2,
+                idlCtx = mkIdlContext(Seq(tok1, tok2, tok3))
+            ),
+            TestDesc(
+                truth = "tok_count() == size(tok_all())",
+                token = tok2,
+                idlCtx = mkIdlContext(Seq(tok1, tok2, tok3))
+            ),
+            TestDesc(
                 truth =
                     "size(tok_all_for_group('g')) == 1 && " +
                     "tok_id(first(tok_all_for_group('w'))) == '2' && " +