def test_type_alias_signature(self, alias, expected_sig):
info_obj = parser.TypeAliasPageInfo(
full_name='tfdocs.api_generator.generate_lib.DocGenerator',
py_object=alias)
info_obj.collect_docs(self.parser_config)
if sys.version_info[:2] <= (3, 6):
# TypeAliasPageInfo.signature is built using the __origin__ attribute of
# type annotations. Before Python 3.7, __origin__ stored typing constructs
# (e.g., typing.List); in 3.7+, it stores the equivalent runtime class
# (e.g., builtins.list).
expected_sig = expected_sig.replace('dict[', 'Dict[').replace(
'list[', 'List[')
# For some reason, bool is missing from the deep_objects signature in 3.6.
expected_sig = expected_sig.replace(' bool,\n', '')
self.assertEqual(info_obj.signature, expected_sig)
def test_type_alias_signature(self, alias, expected_sig):
info_obj = parser.TypeAliasPageInfo(
full_name='tfdocs.api_generator.generate_lib.DocGenerator',
py_object=alias)
info_obj.collect_docs(self.parser_config)
self.assertEqual(info_obj.signature, expected_sig)
