def test_update_thread_substraction_for_loop(): thread = 100 * np.ones(shape=(5, 5)) nails = generate_rectangle_nail_list([0, 0], [4, 4], 4) target = np.array([[100, 100, 80, 100, 100], [100, 100, 90, 50, 100], [100, 100, 90, 50, 100], [100, 100, 90, 100, 50], [100, 100, 90, 100, 100]]) init_distance = distance_image(thread, target) current_nail = [0, 2] distances = [] for i in range(100): if current_nail is None: print(f'breaking at {i}') break current_nail, thread = update_thread_substraction( nails, current_nail, target, thread, 1, distances) new_distance = distance_image(thread, target) print(f'init_distance substraction : {init_distance}') print(f'new_distance substraction : {new_distance}') print(thread) assert new_distance < init_distance
def test_distance_1(): image_1 = np.array([[0, 0], [0, 0]]) image_2 = np.array([[0, 0], [0, 0]]) assert_float_equals(distance_image(image_1, image_2), 0)
def test_distance_5(): image_1 = np.array([[1, 0], [0, 0]]) image_2 = np.array([[0, 2], [0, 0]]) assert_float_equals(distance_image(image_1, image_2), math.sqrt(5))
def test_distance_2(): image_1 = np.array([[0, 0], [0, 0]]) image_2 = np.array([[0, 1], [0, 0]]) assert distance_image(image_1, image_2) == 1