Add switch keywords

releasenotes/notes/switch-case-4ea93b309024bee4.yaml

@@ -0,0 +1,4 @@
+---
+features:
+  - |
+    The new keywords ``switch``, ``case`` and ``default`` now lex correctly.

src/openqasm_pygments/qasm3.py

@@ -92,7 +92,7 @@ def _defcalgrammar_callback(self, match):
             ("let", token.Keyword.Declaration),
             (
                 words(
-                    "break continue end if else for while box return in".split(),
+                    "break continue end if else for while box return in switch case default".split(),
                     prefix="\\b",
                     suffix="\\b",
                 ),

tests/test_qasm3_lexer.py

@@ -85,6 +85,37 @@ def test_annotation_namespace(lexer_qasm3):
     ]
 
 
+def test_switch_case(lexer_qasm3):
+    text = """\
+switch (i) {
+    case 1, 2 {}
+    case 3 {}
+    default {}
+}
+"""
+    assert _remove_whitespace(lexer_qasm3.get_tokens(text)) == [
+        (token.Keyword, "switch"),
+        (token.Punctuation, "("),
+        (token.Name, "i"),
+        (token.Punctuation, ")"),
+        (token.Punctuation, "{"),
+        (token.Keyword, "case"),
+        (token.Literal.Number, "1"),
+        (token.Punctuation, ","),
+        (token.Literal.Number, "2"),
+        (token.Punctuation, "{"),
+        (token.Punctuation, "}"),
+        (token.Keyword, "case"),
+        (token.Literal.Number, "3"),
+        (token.Punctuation, "{"),
+        (token.Punctuation, "}"),
+        (token.Keyword, "default"),
+        (token.Punctuation, "{"),
+        (token.Punctuation, "}"),
+        (token.Punctuation, "}"),
+    ]
+
+
 class TestPulseLexerDelegation:
     def test_inferred_known_alias(self, lexer_qasm3):
         # This uses a very (!) non-standard pulse-grammar lexer to test delegation