black format

python/cuspatial/cuspatial/core/spatial/bound.py

@@ -1,17 +1,16 @@
 # Copyright (c) 2022, NVIDIA CORPORATION.
 
+from cudf import DataFrame
+from cudf.core.column import as_column
+
 from cuspatial._lib.polygon_bounding_boxes import (
     polygon_bounding_boxes as cpp_polygon_bounding_boxes,
 )
 from cuspatial._lib.polyline_bounding_boxes import (
     polyline_bounding_boxes as cpp_polyline_bounding_boxes,
 )
-
 from cuspatial.utils.column_utils import normalize_point_columns
 
-from cudf.core.column import as_column
-from cudf import DataFrame
-
 
 def polygon_bounding_boxes(poly_offsets, ring_offsets, xs, ys):
     """Compute the minimum bounding-boxes for a set of polygons.

python/cuspatial/cuspatial/core/spatial/filter.py

@@ -44,5 +44,7 @@ def points_in_spatial_window(min_x, max_x, min_y, max_y, xs, ys):
     """
     xs, ys = normalize_point_columns(as_column(xs), as_column(ys))
     return DataFrame._from_data(
-        *spatial_window.points_in_spatial_window(min_x, max_x, min_y, max_y, xs, ys)
+        *spatial_window.points_in_spatial_window(
+            min_x, max_x, min_y, max_y, xs, ys
+        )
     )

python/cuspatial/cuspatial/core/spatial/index.py

@@ -11,7 +11,9 @@
 from cuspatial.utils.column_utils import normalize_point_columns
 
 
-def quadtree_on_points(xs, ys, x_min, x_max, y_min, y_max, scale, max_depth, min_size):
+def quadtree_on_points(
+    xs, ys, x_min, x_max, y_min, y_max, scale, max_depth, min_size
+):
     """Construct a quadtree from a set of points for a given area-of-interest
         bounding box.
 

python/cuspatial/cuspatial/core/spatial/join.py

@@ -5,13 +5,12 @@
 from cudf import DataFrame
 from cudf.core.column import as_column
 
+from cuspatial._lib import spatial_join
 from cuspatial._lib.point_in_polygon import (
     point_in_polygon as cpp_point_in_polygon,
 )
-from cuspatial._lib import spatial_join
-
-from cuspatial.utils.column_utils import normalize_point_columns
 from cuspatial.utils import gis_utils
+from cuspatial.utils.column_utils import normalize_point_columns
 
 
 def point_in_polygon(
@@ -242,7 +241,12 @@ def quadtree_point_in_polygon(
             Indices of each point that intersects with a polygon.
     """
 
-    (points_x, points_y, poly_points_x, poly_points_y,) = normalize_point_columns(
+    (
+        points_x,
+        points_y,
+        poly_points_x,
+        poly_points_y,
+    ) = normalize_point_columns(
         as_column(points_x),
         as_column(points_y),
         as_column(poly_points_x),
@@ -313,7 +317,12 @@ def quadtree_point_to_nearest_polyline(
         distance : cudf.Series
             Distances between each point and its nearest polyline.
     """
-    (points_x, points_y, poly_points_x, poly_points_y,) = normalize_point_columns(
+    (
+        points_x,
+        points_y,
+        poly_points_x,
+        poly_points_y,
+    ) = normalize_point_columns(
         as_column(points_x),
         as_column(points_y),
         as_column(poly_points_x),

python/cuspatial/cuspatial/core/spatial/measure.py

@@ -1,20 +1,17 @@
 # Copyright (c) 2022, NVIDIA CORPORATION.
 
-from cuspatial._lib.spatial import (
-    haversine_distance as cpp_haversine_distance,
-)
+from cudf import DataFrame, Series
+from cudf.core.column import as_column
+
 from cuspatial._lib.hausdorff import (
     directed_hausdorff_distance as cpp_directed_hausdorff_distance,
 )
 from cuspatial._lib.linestring_distance import (
     pairwise_linestring_distance as cpp_pairwise_linestring_distance,
 )
-
+from cuspatial._lib.spatial import haversine_distance as cpp_haversine_distance
 from cuspatial.utils.column_utils import normalize_point_columns
 
-from cudf import DataFrame, Series
-from cudf.core.column import as_column
-
 
 def directed_hausdorff_distance(xs, ys, space_offsets):
     """Compute the directed Hausdorff distances between all pairs of
@@ -210,5 +207,9 @@ def pairwise_linestring_distance(offsets1, xs1, ys1, offsets2, xs2, ys2):
     offsets1 = as_column(offsets1, dtype="int32")
     offsets2 = as_column(offsets2, dtype="int32")
     return Series._from_data(
-        {None: cpp_pairwise_linestring_distance(offsets1, xs1, ys1, offsets2, xs2, ys2)}
+        {
+            None: cpp_pairwise_linestring_distance(
+                offsets1, xs1, ys1, offsets2, xs2, ys2
+            )
+        }
     )

python/cuspatial/cuspatial/core/spatial/project.py

@@ -1,6 +1,7 @@
 # Copyright (c) 2022, NVIDIA CORPORATION.
 
 from cudf import DataFrame
+
 from cuspatial._lib.spatial import (
     lonlat_to_cartesian as cpp_lonlat_to_cartesian,
 )

python/cuspatial/cuspatial/tests/spatial/filter/test_points_in_spatial_window.py

@@ -18,16 +18,22 @@ def test_centered():
     result = cuspatial.points_in_spatial_window(
         -1, 1, -1, 1, cudf.Series([0.0]), cudf.Series([0.0])
     )
-    cudf.testing.assert_frame_equal(result, cudf.DataFrame({"x": [0.0], "y": [0.0]}))
+    cudf.testing.assert_frame_equal(
+        result, cudf.DataFrame({"x": [0.0], "y": [0.0]})
+    )
 
 
-@pytest.mark.parametrize("coords", [(-1.0, -1.0), (-1.0, 1.0), (1.0, -1.0), (1.0, 1.0)])
+@pytest.mark.parametrize(
+    "coords", [(-1.0, -1.0), (-1.0, 1.0), (1.0, -1.0), (1.0, 1.0)]
+)
 def test_corners(coords):
     x, y = coords
     result = cuspatial.points_in_spatial_window(
         -1.1, 1.1, -1.1, 1.1, cudf.Series([x]), cudf.Series([y])
     )
-    cudf.testing.assert_frame_equal(result, cudf.DataFrame({"x": [x], "y": [y]}))
+    cudf.testing.assert_frame_equal(
+        result, cudf.DataFrame({"x": [x], "y": [y]})
+    )
 
 
 def test_pair():

python/cuspatial/cuspatial/tests/spatial/join/test_spatial_join.py

@@ -287,7 +287,9 @@ def test_polygon_join_small(dtype):
         intersections,
         cudf.DataFrame(
             {
-                "poly_offset": cudf.Series([3, 3, 1, 2, 1, 1, 0, 3], dtype=np.uint32),
+                "poly_offset": cudf.Series(
+                    [3, 3, 1, 2, 1, 1, 0, 3], dtype=np.uint32
+                ),
                 "quad_offset": cudf.Series(
                     [10, 11, 6, 6, 12, 13, 2, 2], dtype=np.uint32
                 ),
@@ -486,7 +488,9 @@ def test_quadtree_point_in_polygon_small(dtype):
     )
 
 
-def run_test_quadtree_point_to_nearest_polyline_small(dtype, expected_distances):
+def run_test_quadtree_point_to_nearest_polyline_small(
+    dtype, expected_distances
+):
     x_min = 0
     x_max = 8
     y_min = 0

python/cuspatial/cuspatial/tests/spatial/project/test_lonlat_to_cartesian.py

@@ -42,7 +42,9 @@ def test_camera_corners(corner):
     result = cuspatial.lonlat_to_cartesian(
         x[corner], y[corner], cudf.Series(x[corner]), cudf.Series(y[corner])
     )
-    cudf.testing.assert_frame_equal(result, cudf.DataFrame({"x": [0.0], "y": [0.0]}))
+    cudf.testing.assert_frame_equal(
+        result, cudf.DataFrame({"x": [0.0], "y": [0.0]})
+    )
 
 
 def test_longest_distance():
@@ -74,7 +76,9 @@ def test_zeros():
     result = cuspatial.lonlat_to_cartesian(
         0.0, 0.0, cudf.Series([0.0]), cudf.Series([0.0])
     )
-    cudf.testing.assert_frame_equal(result, cudf.DataFrame({"x": [0.0], "y": [0.0]}))
+    cudf.testing.assert_frame_equal(
+        result, cudf.DataFrame({"x": [0.0], "y": [0.0]})
+    )
 
 
 def test_values():