fix: allow "not in" & "is not" to be queried

Author: ObserverOfTime

.github/workflows/ci.yml

@@ -53,6 +53,8 @@ jobs:
             !examples/cpython/Lib/test/test_type_params.py
             !examples/cpython/Lib/test/test_compile.py
             !examples/cpython/Tools/build/generate_re_casefix.py
+            !examples/cpython/Lib/test/test_annotationlib.py
+            !examples/cpython/Lib/test/test_type_params.py
           invalid-files: |
             examples/cpython/Lib/test/tokenizedata/badsyntax_3131.py
   fuzz:

grammar.js

@@ -807,6 +807,10 @@ module.exports = grammar({
       field('argument', $.primary_expression),
     )),
 
+    _not_in: _ => seq('not', 'in'),
+
+    _is_not: _ => seq('is', 'not'),
+
     comparison_operator: $ => prec.left(PREC.compare, seq(
       $.primary_expression,
       repeat1(seq(
@@ -820,9 +824,9 @@ module.exports = grammar({
             '>',
             '<>',
             'in',
-            alias(seq('not', 'in'), 'not in'),
+            alias($._not_in, 'not in'),
             'is',
-            alias(seq('is', 'not'), 'is not'),
+            alias($._is_not, 'is not'),
           )),
         $.primary_expression,
       )),

queries/highlights.scm

@@ -96,6 +96,8 @@
   "is"
   "not"
   "or"
+  "is not"
+  "not in"
 ] @operator
 
 [