fmt

tests/connect/test_intersection.py

@@ -15,7 +15,7 @@ def test_intersection(spark_session):
     # Verify the DataFrame has expected values
     # Intersection should only include overlapping values once
     assert len(results) == 4, "DataFrame should have 4 rows (overlapping values 3,4,5,6)"
-    
+
     # Check that all expected values are present
     values = [row.id for row in results]
     assert sorted(values) == [3, 4, 5, 6], "Values should match expected overlapping sequence"

tests/connect/test_union.py

@@ -15,7 +15,22 @@ def test_union(spark_session):
     # Verify the DataFrame has expected values
     # Union includes duplicates, so length should be sum of both ranges
     assert len(results) == 14, "DataFrame should have 14 rows (7 + 7)"
-    
+
     # Check that all expected values are present, including duplicates
     values = [row.id for row in results]
-    assert sorted(values) == [0, 1, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 8, 9], "Values should match expected sequence with duplicates"
+    assert sorted(values) == [
+        0,
+        1,
+        2,
+        3,
+        3,
+        4,
+        4,
+        5,
+        5,
+        6,
+        6,
+        7,
+        8,
+        9,
+    ], "Values should match expected sequence with duplicates"