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]],
            [[1, 0.5], [0, 0]],
        ],
        tf.float32,
    )
    edge_destination = tf.cast(
        [
            [[1, 1.5], [2, 2]],
            [[1, 1.5], [2, 2]],
        ],
        tf.float32,
    )
    sigma = 1.0
    pafs = edge_maps.make_multi_pafs(
        xv=xv,
        yv=yv,
        edge_sources=edge_source,
        edge_destinations=edge_destination,
        sigma=sigma,
    )
    np.testing.assert_allclose(
        pafs,
        [
            [
                [[0.0, 0.916], [1.414, 1.414]],
                [[0.0, 1.938], [1.248, 1.248]],
                [[0.0, 0.916], [0.191, 0.191]],
            ],
            [
                [[0.0, 1.213], [1.248, 1.248]],
                [[0.0, 2.0], [1.414, 1.414]],
                [[0.0, 1.213], [1.248, 1.248]],
            ],
            [
                [[0.0, 0.916], [0.191, 0.191]],
                [[0.0, 1.938], [1.248, 1.248]],
                [[0.0, 0.916], [1.414, 1.414]],
            ],
        ],
        atol=1e-3,
    )