def main_pyannotate(): try: collect_types.init_types_collection() collect_types.resume() return main() finally: log.info("dumping type_info") collect_types.pause() collect_types.dump_stats('type_info.json')
def collecting_types(self): # type: () -> Iterator[None] collect_types.collected_args = {} collect_types.collected_signatures = {} collect_types.num_samples = {} collect_types.sampling_counters = {} collect_types.call_pending = set() collect_types.resume() yield None collect_types.pause() self.load_stats()
def test_only_return(self): # type: () -> None def only_return(x): # type: (int) -> str collect_types.resume() return '' only_return(1) collect_types.pause() self.load_stats() # No entry is stored if we only have a return event with no matching call. self.assert_type_comments('only_return', [])
def afterTest(self, *args, **kwargs): collect_types.pause()
def collect_types_fixture(): from pyannotate_runtime import collect_types collect_types.resume() yield collect_types.pause()
def pytest_runtest_teardown(self): """Handle the pytest test end hook event: stop type collection.""" collect_types.pause()
def collect_types_fixture(): collect_types.resume() yield collect_types.pause()
#!/usr/bin/env python3 from auth import main from pyannotate_runtime import collect_types if __name__ == '__main__': collect_types.init_types_collection() collect_types.resume() main() collect_types.pause() collect_types.dump_stats('type_info.json')