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)