def METHOD_NAME(post_image_mock, search_results, pysparkling_context):
    mask = np.array([[1, 1], [1, 0]])
    img_id = "iso_image_id"
    post_image_mock.return_value = img_id
    formula_images_rdd = pysparkling_context.parallelize(
        [
            (0, [coo_matrix([[0, 0], [0, 1]]), None, coo_matrix([[2, 3], [1, 0]]), None]),
            (1, [coo_matrix([[1, 1], [0, 1]]), None, None, None]),
        ]
    )
    ids = search_results._post_images_to_image_store(formula_images_rdd, mask, 4)
    assert ids == {
        0: [img_id, None, img_id, None],
        1: [img_id, None, None, None],
    }