fix: hive metadata extractor code lint

Signed-off-by: zhmin <myzhmin@gmail.com>
amundsen-io · Oct 28, 2020 · ce89331 · ce89331
1 parent e882062
commit ce89331
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/tests/unit/extractor/test_hive_table_metadata_extractor.py b/tests/unit/extractor/test_hive_table_metadata_extractor.py
@@ -29,7 +29,7 @@ def test_extraction_with_empty_query_result(self) -> None:
         """
         with patch.object(SQLAlchemyExtractor, '_get_connection'), \
             patch.object(HiveTableMetadataExtractor, '_choose_default_sql_stm',
-                             return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
+                         return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
             extractor = HiveTableMetadataExtractor()
             extractor.init(self.conf)
 
@@ -39,7 +39,7 @@ def test_extraction_with_empty_query_result(self) -> None:
     def test_extraction_with_single_result(self) -> None:
         with patch.object(SQLAlchemyExtractor, '_get_connection') as mock_connection, \
             patch.object(HiveTableMetadataExtractor, '_choose_default_sql_stm',
-                             return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
+                         return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
             connection = MagicMock()
             mock_connection.return_value = connection
             sql_execute = MagicMock()
@@ -107,7 +107,7 @@ def test_extraction_with_single_result(self) -> None:
     def test_extraction_with_multiple_result(self) -> None:
         with patch.object(SQLAlchemyExtractor, '_get_connection') as mock_connection, \
             patch.object(HiveTableMetadataExtractor, '_choose_default_sql_stm',
-                             return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
+                         return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
             connection = MagicMock()
             mock_connection.return_value = connection
             sql_execute = MagicMock()
@@ -248,7 +248,7 @@ def test_sql_statement(self) -> None:
         """
         with patch.object(SQLAlchemyExtractor, '_get_connection'), \
             patch.object(HiveTableMetadataExtractor, '_choose_default_sql_stm',
-                             return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
+                         return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
             extractor = HiveTableMetadataExtractor()
             extractor.init(self.conf)
             self.assertTrue(self.where_clause_suffix in extractor.sql_stmt)
@@ -260,7 +260,7 @@ def test_hive_sql_statement_with_custom_sql(self) -> None:
         """
         with patch.object(SQLAlchemyExtractor, '_get_connection'), \
             patch.object(HiveTableMetadataExtractor, '_choose_default_sql_stm',
-                          return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
+                         return_value=HiveTableMetadataExtractor.DEFAULT_SQL_STATEMENT):
             config_dict = {
                 HiveTableMetadataExtractor.WHERE_CLAUSE_SUFFIX_KEY: self.where_clause_suffix,
                 'extractor.sqlalchemy.{}'.format(SQLAlchemyExtractor.CONN_STRING):
@@ -276,4 +276,3 @@ def test_hive_sql_statement_with_custom_sql(self) -> None:
 
 if __name__ == '__main__':
     unittest.main()
-