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
46 lines
1.3 KiB
Python
46 lines
1.3 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 AccessSpecifier
|
|
from clang.cindex import Cursor
|
|
from clang.cindex import TranslationUnit
|
|
|
|
from .util import get_cursor
|
|
from .util import get_tu
|
|
|
|
import unittest
|
|
|
|
|
|
class TestAccessSpecifiers(unittest.TestCase):
|
|
def test_access_specifiers(self):
|
|
"""Ensure that C++ access specifiers are available on cursors"""
|
|
|
|
tu = get_tu(
|
|
"""
|
|
class test_class {
|
|
public:
|
|
void public_member_function();
|
|
protected:
|
|
void protected_member_function();
|
|
private:
|
|
void private_member_function();
|
|
};
|
|
""",
|
|
lang="cpp",
|
|
)
|
|
|
|
test_class = get_cursor(tu, "test_class")
|
|
self.assertEqual(test_class.access_specifier, AccessSpecifier.INVALID)
|
|
|
|
public = get_cursor(tu.cursor, "public_member_function")
|
|
self.assertEqual(public.access_specifier, AccessSpecifier.PUBLIC)
|
|
|
|
protected = get_cursor(tu.cursor, "protected_member_function")
|
|
self.assertEqual(protected.access_specifier, AccessSpecifier.PROTECTED)
|
|
|
|
private = get_cursor(tu.cursor, "private_member_function")
|
|
self.assertEqual(private.access_specifier, AccessSpecifier.PRIVATE)
|