Fix flow_direction test parametrize: add missing size/dtype params

The cross-backend tests parametrize size and dtype for the random_data
fixture but the test functions didn't declare those parameters,
causing pytest collection to fail.

Author: brendancol

xrspatial/hydro/tests/test_flow_direction_d8.py

@@ -230,7 +230,7 @@ def test_nan_edges():
 @pytest.mark.parametrize("size", [(4, 6), (10, 15)])
 @pytest.mark.parametrize(
     "dtype", [np.int32, np.int64, np.float32, np.float64])
-def test_numpy_equals_dask(random_data):
+def test_numpy_equals_dask(random_data, size, dtype):
     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, flow_direction)
@@ -240,7 +240,7 @@ def test_numpy_equals_dask(random_data):
 @pytest.mark.parametrize("size", [(4, 6), (10, 15)])
 @pytest.mark.parametrize(
     "dtype", [np.int32, np.int64, np.float32, np.float64])
-def test_numpy_equals_cupy(random_data):
+def test_numpy_equals_cupy(random_data, size, dtype):
     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, flow_direction)
@@ -251,7 +251,7 @@ def test_numpy_equals_cupy(random_data):
 @pytest.mark.parametrize("size", [(4, 6), (10, 15)])
 @pytest.mark.parametrize(
     "dtype", [np.int32, np.int64, np.float32, np.float64])
-def test_numpy_equals_dask_cupy(random_data):
+def test_numpy_equals_dask_cupy(random_data, size, dtype):
     numpy_agg = create_test_raster(random_data, backend='numpy')
     dask_cupy_agg = create_test_raster(random_data, backend='dask+cupy')
     assert_numpy_equals_dask_cupy(numpy_agg, dask_cupy_agg, flow_direction,

xrspatial/hydro/tests/test_flow_direction_dinf.py

@@ -361,7 +361,7 @@ def test_dataset_support():
 @pytest.mark.parametrize("size", [(4, 6), (10, 15)])
 @pytest.mark.parametrize(
     "dtype", [np.int32, np.int64, np.float32, np.float64])
-def test_numpy_equals_dask(random_data):
+def test_numpy_equals_dask(random_data, size, dtype):
     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, flow_direction_dinf)
@@ -371,7 +371,7 @@ def test_numpy_equals_dask(random_data):
 @pytest.mark.parametrize("size", [(4, 6), (10, 15)])
 @pytest.mark.parametrize(
     "dtype", [np.int32, np.int64, np.float32, np.float64])
-def test_numpy_equals_cupy(random_data):
+def test_numpy_equals_cupy(random_data, size, dtype):
     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, flow_direction_dinf)
@@ -382,7 +382,7 @@ def test_numpy_equals_cupy(random_data):
 @pytest.mark.parametrize("size", [(4, 6), (10, 15)])
 @pytest.mark.parametrize(
     "dtype", [np.int32, np.int64, np.float32, np.float64])
-def test_numpy_equals_dask_cupy(random_data):
+def test_numpy_equals_dask_cupy(random_data, size, dtype):
     numpy_agg = create_test_raster(random_data, backend='numpy')
     dask_cupy_agg = create_test_raster(random_data, backend='dask+cupy')
     assert_numpy_equals_dask_cupy(numpy_agg, dask_cupy_agg,