Merge pull request #52 from NethermindEth/ElijahVlasov/wrong-annotati…

…on-error-message

Add an error message for unexpected annotations

pyproject.toml

@@ -18,7 +18,7 @@ skip_gitignore = true
 
 [tool.poetry]
 name = "horus-compile"
-version = "0.0.6.9"
+version = "0.0.6.10"
 authors = ["Nethermind <[email protected]>"]
 description = "Use formally verified annotations in your Cairo code"
 classifiers = [

src/horus/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.6.9"
+__version__ = "0.0.6.10"

src/horus/compiler/parser_transformer.py

@@ -51,16 +51,17 @@ def commented_code_element(self, meta, value):
 
         if comment is not None:
             possible_annotation: str = comment.strip()
-
-            for annotation in [
-                "@pre",
-                "@post",
-                "@assert",
-                "@invariant",
-                "@declare",
-                "@storage_update",
-            ]:
-                if possible_annotation.startswith(annotation):
+            if possible_annotation.startswith("@"):
+                annotation_type = possible_annotation.split(" ")[0].strip()
+
+                if annotation_type in [
+                    "@pre",
+                    "@post",
+                    "@assert",
+                    "@invariant",
+                    "@declare",
+                    "@storage_update",
+                ]:
                     check = horus.compiler.parser.parse(
                         filename=self.input_file.filename,
                         code=possible_annotation,
@@ -79,6 +80,11 @@ def commented_code_element(self, meta, value):
                         check, code_elm=code_elem.code_elm
                     )
                     return code_elem
+                else:
+                    raise ParserError(
+                        f"Unexpected annotation type: {annotation_type}",
+                        location=self.meta2loc(meta),
+                    )
 
         return super().commented_code_element(meta, value)