cleanup and formatter

Author: dlawin

tests/test_dbt.py

@@ -1,4 +1,3 @@
-import json
 import os
 
 from data_diff.diff_tables import Algorithm
@@ -11,7 +10,6 @@
     DbtParser,
     RUN_RESULTS_PATH,
     MANIFEST_PATH,
-    PROFILES_FILE,
     PROJECT_FILE,
     DiffVars,
 )
@@ -154,10 +152,7 @@ def test_set_project_dict(self, mock_open, mock_yaml_parse):
 
     def test_set_connection_snowflake_success(self):
         expected_driver = "snowflake"
-        expected_credentials = {
-                "user": "user",
-                "password": "password"
-        }
+        expected_credentials = {"user": "user", "password": "password"}
         mock_self = Mock()
         mock_self._get_connection_creds.return_value = (expected_credentials, expected_driver)
 
@@ -171,9 +166,7 @@ def test_set_connection_snowflake_success(self):
 
     def test_set_connection_snowflake_no_password(self):
         expected_driver = "snowflake"
-        expected_credentials = {
-                "user": "user"
-        }
+        expected_credentials = {"user": "user"}
         mock_self = Mock()
         mock_self._get_connection_creds.return_value = (expected_credentials, expected_driver)
 
@@ -185,10 +178,10 @@ def test_set_connection_snowflake_no_password(self):
     def test_set_connection_bigquery_success(self):
         expected_driver = "bigquery"
         expected_credentials = {
-                        "method": "oauth",
-                        "project": "a_project",
-                        "dataset": "a_dataset",
-                    }
+            "method": "oauth",
+            "project": "a_project",
+            "dataset": "a_dataset",
+        }
         mock_self = Mock()
         mock_self._get_connection_creds.return_value = (expected_credentials, expected_driver)
 
@@ -202,10 +195,10 @@ def test_set_connection_bigquery_success(self):
     def test_set_connection_bigquery_not_oauth(self):
         expected_driver = "bigquery"
         expected_credentials = {
-                        "method": "not_oauth",
-                        "project": "a_project",
-                        "dataset": "a_dataset",
-                    }
+            "method": "not_oauth",
+            "project": "a_project",
+            "dataset": "a_dataset",
+        }
 
         mock_self = Mock()
         mock_self._get_connection_creds.return_value = (expected_credentials, expected_driver)
@@ -300,7 +293,6 @@ def test_get_connection_no_credentials(self, mock_file):
         with self.assertRaises(ValueError):
             _, _ = DbtParser._get_connection_creds(mock_self)
 
-
     profile_yaml_no_target_credentials = """
     a_profile:
       outputs:
@@ -336,6 +328,7 @@ def test_get_connection_no_type(self, mock_file):
         with self.assertRaises(ValueError):
             _, _ = DbtParser._get_connection_creds(mock_self)
 
+
 class TestDbtDiffer(unittest.TestCase):
     # These two integration tests can be used to test a real diff
     # export DATA_DIFF_DBT_PROJ=/path/to/a/dbt/project