def METHOD_NAME(size):
    da = make_histogram2d()
    result = median_filter(da, size=size)
    reference = da.copy()
    reference.values = scipy.ndimage.median_filter(
        reference.values, size=size if isinstance(size, int) else size.values()
    )
    assert sc.identical(result, reference)