Improving regexs that look for attributes

VUnit · May 31, 2022 · e21bacf · e21bacf
1 parent fbb46c6
commit e21bacf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vunit/test/bench.py b/vunit/test/bench.py
@@ -559,8 +559,8 @@ def associate(attr):
 
 
 _RE_ATTR_NAME = r"[a-zA-Z0-9_\-]+"
-_RE_ATTRIBUTE = re.compile(r"vunit:\s*(?P<name>\.?" + _RE_ATTR_NAME + r")", re.IGNORECASE)
-_RE_PRAGMA_LEGACY = re.compile(r"vunit_pragma\s+(?P<name>" + _RE_ATTR_NAME + ")", re.IGNORECASE)
+_RE_ATTRIBUTE = re.compile(r"(?:--|\/\/)\s*vunit:\s*(?P<name>\.?" + _RE_ATTR_NAME + r")", re.IGNORECASE)
+_RE_PRAGMA_LEGACY = re.compile(r"(?:--|\/\/)\s*vunit_pragma\s+(?P<name>" + _RE_ATTR_NAME + r")", re.IGNORECASE)
 _VALID_ATTRIBUTES = ["run_all_in_same_sim", "fail_on_warning"]