add Token.matches(*, name, str)

asottile · Jul 30, 2023 · dcfe4e9 · dcfe4e9
1 parent 09fb26d
commit dcfe4e9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/tokenize_rt_test.py b/tests/tokenize_rt_test.py
@@ -37,6 +37,14 @@ def test_token_offset():
     assert token.offset == Offset(line=1, utf8_byte_offset=2)
 
 
+def test_token_matches():
+    token = Token('NAME', 'x', line=1, utf8_byte_offset=2)
+    assert token.matches(name='NAME', src='x')
+    assert not token.matches(name='OP', src='x')
+    assert not token.matches(name='NAME', src='y')
+    assert not token.matches(name='OP', src=':')
+
+
 def test_src_to_tokens_simple():
     src = 'x = 5\n'
     ret = src_to_tokens(src)

diff --git a/tokenize_rt.py b/tokenize_rt.py
@@ -40,6 +40,9 @@ class Token(NamedTuple):
     def offset(self) -> Offset:
         return Offset(self.line, self.utf8_byte_offset)
 
+    def matches(self, *, name: str, src: str) -> bool:
+        return self.name == name and self.src == src
+
 
 _string_re = re.compile('^([^\'"]*)(.*)$', re.DOTALL)
 _escaped_nl_re = re.compile(r'\\(\n|\r\n|\r)')