Fix formatting; add tests

Author: kesmit13

src/sql/connection/connection.py

@@ -1,6 +1,7 @@
 import warnings
 import difflib
 import abc
+import operator as op
 import os
 from difflib import get_close_matches
 import atexit
@@ -112,10 +113,16 @@ def __init__(self) -> None:
         self._result_sets = []
 
     def append(self, result):
-        map(self._result_sets.pop, reversed([
-            i for i, item in enumerate(self._result_sets)
-            if len(result) == len(item) and map(op.eq, zip(result, item))
-        ]))
+        map(
+            self._result_sets.pop,
+            reversed(
+                [
+                    i
+                    for i, item in enumerate(self._result_sets)
+                    if len(result) == len(item) and map(op.eq, zip(result, item))
+                ]
+            ),
+        )
 
         self._result_sets.append(result)
 

src/tests/test_connection.py

@@ -1073,6 +1073,25 @@ def test_result_set_collection_append():
     assert collection._result_sets == [1, 2]
 
 
+def test_result_set_collection_append_numpy():
+    try:
+        import numpy as np
+
+        a1 = np.array([1, 2])
+        a2 = np.array([3, 4])
+
+        collection = ResultSetCollection()
+        collection.append(a1)
+        collection.append(a2)
+
+        assert len(collection._result_sets) == 2
+        assert collection._result_sets[0] is a1
+        assert collection._result_sets[1] is a2
+
+    except ImportError:
+        pass
+
+
 def test_result_set_collection_iterate():
     collection = ResultSetCollection()
     collection.append(1)