def METHOD_NAME():
    xv, yv = make_grid_vectors(image_height=3, image_width=3, output_stride=1)
    edge_source = tf.cast([[1, 0.5], [0, 0]], tf.float32)
    edge_destination = tf.cast([[1, 1.5], [2, 2]], tf.float32)
    sigma = 1.0
    edge_confidence_map = edge_maps.make_edge_maps(
        xv=xv,
        yv=yv,
        edge_source=edge_source,
        edge_destination=edge_destination,
        sigma=sigma,
    )
    np.testing.assert_allclose(
        edge_confidence_map,
        [
            [[0.458, 1.000], [0.969, 0.882], [0.458, 0.135]],
            [[0.607, 0.882], [1.000, 1.000], [0.607, 0.882]],
            [[0.458, 0.135], [0.969, 0.882], [0.458, 1.000]],
        ],
        atol=1e-3,
    )