28 lines · python
1import os2 3from clang.cindex import Config, LanguageKind4 5if "CLANG_LIBRARY_PATH" in os.environ:6 Config.set_library_path(os.environ["CLANG_LIBRARY_PATH"])7 8import unittest9 10from .util import get_cursor, get_tu11 12 13class TestCursorLanguage(unittest.TestCase):14 def test_c(self):15 tu = get_tu("int a;", lang="c")16 main_func = get_cursor(tu.cursor, "a")17 self.assertEqual(main_func.language, LanguageKind.C)18 19 def test_c(self):20 tu = get_tu("class Cls {};", lang="cpp")21 main_func = get_cursor(tu.cursor, "Cls")22 self.assertEqual(main_func.language, LanguageKind.C_PLUS_PLUS)23 24 def test_obj_c(self):25 tu = get_tu("@interface If : NSObject", lang="objc")26 main_func = get_cursor(tu.cursor, "If")27 self.assertEqual(main_func.language, LanguageKind.OBJ_C)28