Add scope keyword for variable definitions

OdysseasKr · Nov 14, 2024 · f0a6b68 · f0a6b68
1 parent 15f1e49
commit f0a6b68
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/rapidchecker/parser/grammar.py b/rapidchecker/parser/grammar.py
@@ -60,12 +60,13 @@
 
 # Variable definition
 var_def = (
-    (T.PERS | T.VAR | T.CONST)
+    pp.Optional(T.LOCAL | T.TASK)
+    + (T.PERS | T.VAR | T.CONST)
     + datatype
-    + identifier
-    + pp.Optional("{" + expression + "}")
-    + pp.Optional(":=" + expression)
-    + ";"
+    - identifier
+    - pp.Optional("{" + expression + "}")
+    - pp.Optional(":=" + expression)
+    - ";"
 )
 var_def_section = pp.ZeroOrMore(var_def)
 

diff --git a/rapidchecker/parser/tokens.py b/rapidchecker/parser/tokens.py
@@ -56,6 +56,8 @@ def __init__(self, *args, **kwargs) -> None:
 DEFAULT = RapidKeyword("DEFAULT")
 ERROR = RapidKeyword("ERROR")
 INOUT = RapidKeyword("INOUT")
+TASK = RapidKeyword("TASK")
+LOCAL = RapidKeyword("LOCAL")
 
 RESERVED_WORD = (
     MODULE
@@ -98,6 +100,9 @@ def __init__(self, *args, **kwargs) -> None:
     | CASE
     | DEFAULT
     | ERROR
+    | INOUT
+    | TASK
+    | LOCAL
 )
 
 MODULE_OPTIONS = oneOf("SYSMODULE NOSTEPIN VIEWONLY READONLY")
diff --git a/tests/test_grammar.py b/tests/test_grammar.py
@@ -154,6 +154,7 @@ def test_record_def(input_str: str) -> None:
         "VAR robtarget targets{1000};",
         "VAR robtarget targets{var1 + var2};",
         "CONST num number := 1 + 1;",
+        "LOCAL CONST num number := 1 + 1;",
     ],
 )
 def test_var_def(input_str: str) -> None: