def test_gen_global_max(): ''' Test global maximum calculation. ''' from planarity.reference import _elementwise_max rows, cols = 32, 32 image = gen_random_image(rows, cols) image2 = gen_random_image(rows, cols) args = {'in_ptr_1':0, 'in_ptr_2':block_size[0]*block_size[1]} test_res, _ = run_codegen_function(image, gen_global_max, block_size, args, 1, image2buffer={1:image2}) ref_res = _elementwise_max(image, image2) assert ref_res == test_res
def test_elementwise_max(): m1 = [ [1, 2, 3, 4], [-1, 0, 2, 5], [4, 6, 2, 1]] m2 = [ [7, 3,-9, 3], [-4, 2, 0, 1], [0, 8, 2, 7]] test_res = _elementwise_max(m1, m2) ref_res = [ [7, 3, 3, 4], [-1, 2, 2, 5], [4, 8, 2, 7]] assert ref_res == test_res