30 lines · python
1import os2 3from clang.cindex import Config, TokenKind4 5if "CLANG_LIBRARY_PATH" in os.environ:6 Config.set_library_path(os.environ["CLANG_LIBRARY_PATH"])7 8import unittest9 10 11class TestTokenKind(unittest.TestCase):12 def test_registration(self):13 """Ensure that items registered appear as class attributes."""14 self.assertTrue(hasattr(TokenKind, "LITERAL"))15 literal = TokenKind.LITERAL16 17 self.assertIsInstance(literal, TokenKind)18 19 def test_from_value(self):20 """Ensure registered values can be obtained from from_value()."""21 t = TokenKind.from_value(3)22 self.assertIsInstance(t, TokenKind)23 self.assertEqual(t, TokenKind.LITERAL)24 25 def test_repr(self):26 """Ensure repr() works."""27 28 r = repr(TokenKind.LITERAL)29 self.assertEqual(r, "TokenKind.LITERAL")30