drgn/tests/test_python.py
Omar Sandoval 181ebe1a01 Add missing entries in drgn.__all__
PlatformFlags and PrimitiveType got squashed into one string because of
a missing comma, and execscript was never added. Fix it and add some
test cases for it.
2019-10-03 16:50:00 -07:00

23 lines
837 B
Python

import _drgn
import drgn
import unittest
class TestModule(unittest.TestCase):
def test_all(self):
# At least for now, everything in the Python library should go in
# __all__, so make sure that happens.
from_python = {
name for name in dir(drgn)
if not name.startswith('_')
and getattr(getattr(drgn, name), '__module__', '').startswith('drgn')
}
self.assertEqual(from_python - set(drgn.__all__), set())
def test_bindings(self):
# Make sure everything in the C extension (_drgn) is added to the
# Python library (drgn).
from_extension = {name for name in dir(_drgn) if not name.startswith('_')}
self.assertEqual(from_extension - set(dir(drgn)), set())
self.assertEqual(from_extension - set(drgn.__all__), set())