chore(cursor): lint

Author: Brooke-white

redshift_connector/cursor.py

@@ -262,14 +262,10 @@ def executemany(self: "Cursor", operation, param_sets) -> "Cursor":
         self._redshift_row_count = -1 if -1 in redshift_rowcounts else sum(rowcounts)
         return self
 
-    def insert_data_bulk(
-        self: "Cursor", filename, table_name, column_indexes, column_names, delimeter
-    ) -> "Cursor":
+    def insert_data_bulk(self: "Cursor", filename, table_name, column_indexes, column_names, delimeter) -> "Cursor":
 
         """runs a single bulk insert statement into the database.
-
         This method is native to redshift_connector.
-
          :param filename: str
              The name of the file to read from.
          :param table_name: str
@@ -280,21 +276,14 @@ def insert_data_bulk(
              The indexes of the columns in the table to insert to.
          :param delimeter: str
              The delimeter to use when reading the file.
-
          Returns
-
          -------
          The Cursor object used for executing the specified database operation: :class:`Cursor`
-
         """
         if not self.__is_valid_table(table_name):
-            raise InterfaceError(
-                "Invalid table name passed to insert_data_bulk: {}".format(table_name)
-            )
+            raise InterfaceError("Invalid table name passed to insert_data_bulk: {}".format(table_name))
         if not self.__has_valid_columns(table_name, column_names):
-            raise InterfaceError(
-                "Invalid column names passed to insert_data_bulk: {}".format(table_name)
-            )
+            raise InterfaceError("Invalid column names passed to insert_data_bulk: {}".format(table_name))
         orig_paramstyle = self.paramstyle
         import csv
 
@@ -325,16 +314,12 @@ def insert_data_bulk(
 
         return self
 
-    def __has_valid_columns(
-        self: "Cursor", table: str, columns: typing.List[str]
-    ) -> bool:
+    def __has_valid_columns(self: "Cursor", table: str, columns: typing.List[str]) -> bool:
         split_table_name: typing.List[str] = table.split(".")
         q: str = "select 1 from information_schema.columns where table_name = ? and column_name = ?"
         if len(split_table_name) == 2:
             q += " and table_schema = ?"
-            param_list = [
-                [split_table_name[1], c, split_table_name[0]] for c in columns
-            ]
+            param_list = [[split_table_name[1], c, split_table_name[0]] for c in columns]
         else:
             param_list = [[split_table_name[0], c] for c in columns]
         temp = self.paramstyle
@@ -344,11 +329,7 @@ def __has_valid_columns(
                 self.execute(q, params)
                 res = self.fetchone()
                 if typing.cast(typing.List[int], res)[0] != 1:
-                    raise InterfaceError(
-                        "Invalid column name: {} specified for table: {}".format(
-                            params[1], table
-                        )
-                    )
+                    raise InterfaceError("Invalid column name: {} specified for table: {}".format(params[1], table))
         except:
             raise
         finally:

test/unit/test_cursor.py

@@ -1,6 +1,6 @@
 import typing
 from test.utils import pandas_only
-from unittest.mock import Mock, PropertyMock, patch ,mock_open
+from unittest.mock import Mock, PropertyMock, mock_open, patch
 
 import pytest  # type: ignore
 
@@ -262,9 +262,7 @@ def test_insert_data_column_names_indexes_mismatch_raises(indexes, names, mocker
     mock_cursor._c = Mock()
     mock_cursor.paramstyle = "qmark"
 
-    with pytest.raises(
-        InterfaceError, match="Column names and indexes must be the same length"
-    ):
+    with pytest.raises(InterfaceError, match="Column names and indexes must be the same length"):
         mock_cursor.insert_data_bulk(
             filename="test_file",
             table_name="test_table",