from copapy import stencil_database, stencil_db, get_local_arch


def test_list_symbols():
    arch = get_local_arch()
    sdb = stencil_database(f'src/copapy/obj/stencils_{arch}_O3.o')
    print('----')
    #print(sdb.function_definitions)
    for sym_name in sdb.function_definitions.keys():
        print('\n-', sym_name)
        print(list(sdb.get_patch_positions(sym_name)))


def test_start_end_function():
    arch = get_local_arch()
    sdb = stencil_database(f'src/copapy/obj/stencils_{arch}_O3.o')
    for sym_name in sdb.function_definitions.keys():
        data = sdb.elf.symbols[sym_name].data
        print('-', sym_name, stencil_db.get_stencil_position(data, sdb.elf.byteorder), len(data))

        start, end = stencil_db.get_stencil_position(data, sdb.elf.byteorder)

        assert start >= 0 and end >= start and end <= len(data)


if __name__ == "__main__":
    test_list_symbols()
    test_start_end_function()