dd3c26a045
This is an ongoing series of commits that are reformatting our Python code. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: MatzeB Differential Revision: https://reviews.llvm.org/D150761
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
import os
|
|
from clang.cindex import Config
|
|
|
|
if "CLANG_LIBRARY_PATH" in os.environ:
|
|
Config.set_library_path(os.environ["CLANG_LIBRARY_PATH"])
|
|
|
|
from clang.cindex import TokenKind
|
|
|
|
import unittest
|
|
|
|
|
|
class TestTokenKind(unittest.TestCase):
|
|
def test_constructor(self):
|
|
"""Ensure TokenKind constructor works as expected."""
|
|
|
|
t = TokenKind(5, "foo")
|
|
|
|
self.assertEqual(t.value, 5)
|
|
self.assertEqual(t.name, "foo")
|
|
|
|
def test_bad_register(self):
|
|
"""Ensure a duplicate value is rejected for registration."""
|
|
|
|
with self.assertRaises(ValueError):
|
|
TokenKind.register(2, "foo")
|
|
|
|
def test_unknown_value(self):
|
|
"""Ensure trying to fetch an unknown value raises."""
|
|
|
|
with self.assertRaises(ValueError):
|
|
TokenKind.from_value(-1)
|
|
|
|
def test_registration(self):
|
|
"""Ensure that items registered appear as class attributes."""
|
|
self.assertTrue(hasattr(TokenKind, "LITERAL"))
|
|
literal = TokenKind.LITERAL
|
|
|
|
self.assertIsInstance(literal, TokenKind)
|
|
|
|
def test_from_value(self):
|
|
"""Ensure registered values can be obtained from from_value()."""
|
|
t = TokenKind.from_value(3)
|
|
self.assertIsInstance(t, TokenKind)
|
|
self.assertEqual(t, TokenKind.LITERAL)
|
|
|
|
def test_repr(self):
|
|
"""Ensure repr() works."""
|
|
|
|
r = repr(TokenKind.LITERAL)
|
|
self.assertEqual(r, "TokenKind.LITERAL")
|