|
6 | 6 |
|
7 | 7 | from reproject import reproject_interp |
8 | 8 | from reproject.hips import reproject_to_hips |
9 | | -from reproject.hips._dask_array import hips_as_dask_and_wcs |
| 9 | +from reproject.hips._dask_array import hips_as_dask_array |
10 | 10 |
|
11 | 11 |
|
12 | 12 | class TestHIPSDaskArray: |
@@ -37,7 +37,7 @@ def test_roundtrip(self, tmp_path, frame, level): |
37 | 37 | ) |
38 | 38 |
|
39 | 39 | # Represent the HiPS as a dask array |
40 | | - dask_array, wcs = hips_as_dask_and_wcs(output_directory, level=level) |
| 40 | + dask_array, wcs = hips_as_dask_array(output_directory, level=level) |
41 | 41 |
|
42 | 42 | # Reproject back to the original WCS |
43 | 43 | final_array, footprint = reproject_interp( |
@@ -73,17 +73,17 @@ def test_level_validation(self, tmp_path): |
73 | 73 | tile_size=32, |
74 | 74 | ) |
75 | 75 |
|
76 | | - dask_array, wcs = hips_as_dask_and_wcs(output_directory, level=0) |
| 76 | + dask_array, wcs = hips_as_dask_array(output_directory, level=0) |
77 | 77 | assert dask_array.shape == (160, 160) |
78 | 78 |
|
79 | | - dask_array, wcs = hips_as_dask_and_wcs(output_directory, level=1) |
| 79 | + dask_array, wcs = hips_as_dask_array(output_directory, level=1) |
80 | 80 | assert dask_array.shape == (320, 320) |
81 | 81 |
|
82 | | - dask_array, wcs = hips_as_dask_and_wcs(output_directory) |
| 82 | + dask_array, wcs = hips_as_dask_array(output_directory) |
83 | 83 | assert dask_array.shape == (320, 320) |
84 | 84 |
|
85 | 85 | with pytest.raises(Exception, match=r"does not contain level 2 data"): |
86 | | - hips_as_dask_and_wcs(output_directory, level=2) |
| 86 | + hips_as_dask_array(output_directory, level=2) |
87 | 87 |
|
88 | 88 | with pytest.raises(Exception, match=r"should be positive"): |
89 | | - hips_as_dask_and_wcs(output_directory, level=-1) |
| 89 | + hips_as_dask_array(output_directory, level=-1) |
0 commit comments