copapy/tests/test_stencil_db.py

30 lines
936 B
Python

from copapy import stencil_database, stencil_db
import platform
def test_list_symbols():
arch = platform.machine()
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 = platform.machine()
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()