Ran black

Author: erezsh

data_diff/databases/base.py

@@ -187,7 +187,9 @@ def query_table_schema(self, path: DbPath) -> Dict[str, tuple]:
  assert len(d) == len(rows)
  return d
 
- def _process_table_schema(self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str], where: str = None):
+ def _process_table_schema(
+ self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str], where: str = None
+ ):
  accept = {i.lower() for i in filter_columns}
 
  col_dict = {row[0]: self._parse_type(path, *row) for name, row in raw_schema.items() if name.lower() in accept}

data_diff/databases/database_types.py

@@ -177,7 +177,9 @@ def query_table_schema(self, path: DbPath) -> Dict[str, tuple]:
  ...
 
  @abstractmethod
- def _process_table_schema(self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str], where: str = None):
+ def _process_table_schema(
+ self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str], where: str = None
+ ):
  """Process the result of query_table_schema().
 
  Done in a separate step, to minimize the amount of processed columns.

data_diff/databases/databricks.py

@@ -83,7 +83,9 @@ def query_table_schema(self, path: DbPath) -> Dict[str, tuple]:
  assert len(d) == len(rows)
  return d
 
- def _process_table_schema(self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str], where: str = None):
+ def _process_table_schema(
+ self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str], where: str = None
+ ):
  accept = {i.lower() for i in filter_columns}
  rows = [row for name, row in raw_schema.items() if name.lower() in accept]
 

tests/test_diff_tables.py

@@ -444,13 +444,13 @@ def test_string_keys(self):
  self.assertRaises(ValueError, list, differ.diff_tables(self.a, self.b))
 
  def test_where_sampling(self):
- a = self.a.replace(where='1=1')
+ a = self.a.replace(where="1=1")
 
  differ = TableDiffer()
  diff = list(differ.diff_tables(a, self.b))
  self.assertEqual(diff, [("-", (str(self.new_uuid), "This one is different"))])
 
- a_empty = self.a.replace(where='1=0')
+ a_empty = self.a.replace(where="1=0")
  self.assertRaises(ValueError, list, differ.diff_tables(a_empty, self.b))