def test_apply_dask_numpy(data_apply):
data, kernel, expected_result = data_apply
dask_numpy_agg = create_test_raster(data, backend='dask')
dask_numpy_apply = apply(dask_numpy_agg, kernel, func_zero_cpu)
general_output_checks(dask_numpy_agg, dask_numpy_apply, expected_result)
def test_hotspots_numpy(data_hotspots):
data, kernel, expected_result = data_hotspots
numpy_agg = create_test_raster(data)
numpy_hotspots = hotspots(numpy_agg, kernel)
general_output_checks(numpy_agg, numpy_hotspots, expected_result)
def test_hillshade_gpu_equals_cpu(random_data):
numpy_agg = create_test_raster(random_data, backend='numpy')
cupy_agg = create_test_raster(random_data, backend='cupy')
assert_numpy_equals_cupy(numpy_agg, cupy_agg, hillshade, rtol=1e-6)
def test_apply_numpy(data_apply):
data, kernel, expected_result = data_apply
numpy_agg = create_test_raster(data)
numpy_apply = apply(numpy_agg, kernel, func_zero_cpu)
general_output_checks(numpy_agg, numpy_apply, expected_result)
def test_calc_cellsize_no_attrs(convolve_2d_data):
agg = create_test_raster(convolve_2d_data)
cellsize = calc_cellsize(agg)
assert cellsize == (1, 1)
def test_hillshade_numpy_equals_dask_numpy(random_data):
numpy_agg = create_test_raster(random_data, backend='numpy')
dask_agg = create_test_raster(random_data, backend='dask')
assert_numpy_equals_dask_numpy(numpy_agg, dask_agg, hillshade)
def data_values_2d(backend):
data = np.asarray([[0, 0, 1, 1, 2, 2, 3, np.inf],
[0, 0, 1, 1, 2, np.nan, 3, 0],
[np.inf, 0, 1, 1, 2, 2, 3, 3]])
agg = create_test_raster(data, backend)
return agg
def test_calc_cellsize_unit_input_attrs(convolve_2d_data):
agg = create_test_raster(convolve_2d_data, attrs={'res': 1, 'unit': 'km'})
cellsize = calc_cellsize(agg)
assert cellsize == (1000, 1000)
def data_zones(backend):
data = np.array([[0, 0, 1, 1, 2, 2, 3, 3], [0, 0, 1, 1, 2, 2, 3, 3],
[0, 0, 1, 1, 2, np.nan, 3, 3]])
agg = create_test_raster(data, backend)
return agg
def test_numpy_equals_cupy_random_data(random_data):
numpy_agg = create_test_raster(random_data, backend='numpy')
cupy_agg = create_test_raster(random_data, backend='cupy')
assert_numpy_equals_cupy(numpy_agg, cupy_agg, aspect, atol=1e-6, rtol=1e-6)
def test_numpy_equals_dask_random_data(random_data):
numpy_agg = create_test_raster(random_data, backend='numpy')
dask_agg = create_test_raster(random_data, backend='dask')
assert_numpy_equals_dask_numpy(numpy_agg, dask_agg, aspect)
def test_numpy_equals_cupy_random_data(random_data):
numpy_agg = create_test_raster(random_data, backend='numpy')
cupy_agg = create_test_raster(random_data, backend='cupy')
assert_numpy_equals_cupy(numpy_agg, cupy_agg, curvature)
def input_data():
data = np.array([[0, 1, 0, 0], [1, 1, 0, 0], [0, 1, 2, 2], [1, 0, 2, 0],
[0, 2, 2, 2]])
agg = create_test_raster(data, dims=['lat', 'lon'])
return agg