Function get_arange_img

tests/test_utils.py:434–441

Returns an image as a numpy array (complete with channel as dim 0) with contents that iterate like an arange.

(size, dtype=np.float32, offset=0)

def get_arange_img(size, dtype=np.float32, offset=0):
    """
    Returns an image as a numpy array (complete with channel as dim 0)
    with contents that iterate like an arange.
    """
    n_elem = np.prod(size)
    img = np.arange(offset, offset + n_elem, dtype=dtype).reshape(size)
    return np.expand_dims(img, 0)


class DistTestCase(unittest.TestCase):