Function test_rewrite

dask/array/tests/test_atop.py:379–399

(inputs, expected)

    ],
)
def test_rewrite(inputs, expected):
    inputs = [
        Blockwise(
            *inp, numblocks={k: (1,) * len(v) for k, v in inp[-1] if v is not None}
        )
        for inp in inputs
    ]
    result = rewrite_blockwise(inputs)

    result2 = (
        result.output,
        "".join(result.output_indices),
        result.task,
        [
            (name, "".join(ind) if ind is not None else ind)
            for name, ind in result.indices
        ],
    )
    # Assert on the task first to get a more informative error message
    assert result2[2] == expected[2]
    assert result2 == expected


def test_index_subs():