class TestFormatAnnotation(unittest.TestCase):
def test_typing_replacement(self):
- from test.typinganndata.ann_module9 import ann, ann1
+ from test.typinganndata.ann_module9 import A, ann, ann1
self.assertEqual(inspect.formatannotation(ann), 'Union[List[str], int]')
self.assertEqual(inspect.formatannotation(ann1), 'Union[List[testModule.typing.A], int]')
+ self.assertEqual(inspect.formatannotation(A, 'testModule.typing'), 'A')
+ self.assertEqual(inspect.formatannotation(A, 'other'), 'testModule.typing.A')
+ self.assertEqual(
+ inspect.formatannotation(ann1, 'testModule.typing'),
+ 'Union[List[testModule.typing.A], int]',
+ )
+
+ def test_formatannotationrelativeto(self):
+ from test.typinganndata.ann_module9 import A, ann1
+
+ # Builtin types:
+ self.assertEqual(
+ inspect.formatannotationrelativeto(object)(type),
+ 'type',
+ )
+
+ # Custom types:
+ self.assertEqual(
+ inspect.formatannotationrelativeto(None)(A),
+ 'testModule.typing.A',
+ )
+
+ class B: ...
+ B.__module__ = 'testModule.typing'
+
+ self.assertEqual(
+ inspect.formatannotationrelativeto(B)(A),
+ 'A',
+ )
+
+ self.assertEqual(
+ inspect.formatannotationrelativeto(object)(A),
+ 'testModule.typing.A',
+ )
+
+ # Not an instance of "type":
+ self.assertEqual(
+ inspect.formatannotationrelativeto(A)(ann1),
+ 'Union[List[testModule.typing.A], int]',
+ )
+
class TestIsMethodDescriptor(unittest.TestCase):