Function _test_4751

tests/test_memory.py:240–319

()

def _test_4751():
    import gc
    import tracemalloc
    
    def analysis(stream_data, do_iter=True):
        pdf_info = pymupdf.Document(stream=stream_data, filetype='pdf')
        tmp_list = range(len(pdf_info))
        for page_num in tmp_list:
            page = pdf_info[page_num]
            raw_info = page.get_text('rawdict')['blocks']
            page_widgets_list = page.widgets() 
            if do_iter:
                for widget_info in page_widgets_list:
                    print(widget_info)
            del page_widgets_list
        pdf_info.close()
        pdf_info = None
        pymupdf.TOOLS.store_shrink(100)

    file_path = os.path.normpath(f'{__file__}/../../tests/resources/test_4751.pdf')
    
    def log(text):
        print(text, flush=1)

    # We filter out all allocations where leaf-most frame is in tracemalloc
    # itself, or in test_memory.py itself, because these are not relevant
    # to finding leaks in pymupdf.
    #
    tm_filters = [
            tracemalloc.Filter(inclusive=False, filename_pattern=tracemalloc.__file__, all_frames=True),
            tracemalloc.Filter(inclusive=False, filename_pattern=__file__),
            ]

    def get_snapshot():
        '''
        Wrapper for tracemalloc.take_snapshot() that filters out blocks with
        backtraces that we are not interested in.
        '''
        ret = tracemalloc.take_snapshot()
        ret2 = ret.filter_traces(tm_filters)
        #log(f'    {len(ret.traces)=} => {len(ret2.traces)=}')
        return ret2

    # Check that `analysis()` does not leak.
    #
    num_leaks = 0
    with open(file_path,'rb') as f:
        bytes_data = f.read()
    
    tracemalloc.start(30)
    snapshot_prev = get_snapshot()
    
    for it in range(2):
        log('')
        log(f'{it=}')
        
        current, peak = tracemalloc.get_traced_memory()
        log(f'    {current=} {peak=}')