From 6f43372608fc3e50cb5736dc7fab5101c556f312 Mon Sep 17 00:00:00 2001
From: Andrew Gazelka <andrew.gazelka@gmail.com>
Date: Thu, 5 Dec 2024 13:51:45 -0800
Subject: [PATCH] fmt

---
 tests/connect/test_intersection.py |  2 +-
 tests/connect/test_union.py        | 19 +++++++++++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/tests/connect/test_intersection.py b/tests/connect/test_intersection.py
index 7944de5cae..200f391f39 100644
--- a/tests/connect/test_intersection.py
+++ b/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"
diff --git a/tests/connect/test_union.py b/tests/connect/test_union.py
index 9ac235d9e5..34157fd2c1 100644
--- a/tests/connect/test_union.py
+++ b/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"