recce-nightly 1.3.0.20250507__py3-none-any.whl → 1.4.0.20250515__py3-none-any.whl

tests/adapter/dbt_adapter/dbt_test_helper.py

@@ -4,8 +4,18 @@ import uuid
 from datetime import datetime
 from io import StringIO
 
-from dbt.contracts.graph.nodes import ModelNode, SnapshotNode, SeedNode, SourceDefinition
-from dbt.contracts.results import CatalogArtifact, ColumnMetadata, CatalogTable, TableMetadata
+from dbt.contracts.graph.nodes import (
+    ModelNode,
+    SeedNode,
+    SnapshotNode,
+    SourceDefinition,
+)
+from dbt.contracts.results import (
+    CatalogArtifact,
+    CatalogTable,
+    ColumnMetadata,
+    TableMetadata,
+)
 
 from recce.adapter.dbt_adapter import DbtAdapter, as_manifest, load_manifest
 from recce.core import RecceContext
@@ -19,9 +29,9 @@ class DbtTestHelper:
         self.curr_schema = f"{schema_prefix}_curr"
 
         current_dir = os.path.dirname(os.path.abspath(__file__))
-        project_dir = os.path.join(current_dir, 'test_proj')
+        project_dir = os.path.join(current_dir, "test_proj")
         profiles_dir = project_dir
-        manifest_path = os.path.join(project_dir, 'manifest.json')
+        manifest_path = os.path.join(project_dir, "manifest.json")
 
         dbt_adapter = DbtAdapter.load(
             no_artifacts=True,
@@ -30,7 +40,7 @@ class DbtTestHelper:
         )
 
         context = RecceContext()
-        context.adapter_type = 'dbt'
+        context.adapter_type = "dbt"
         context.adapter = dbt_adapter
         context.schema_prefix = schema_prefix
         self.adapter = dbt_adapter
@@ -108,16 +118,16 @@ class DbtTestHelper:
             if csv:
                 dbt_adapter = self.adapter
                 csv = textwrap.dedent(csv)
-                with dbt_adapter.connection_named('create model'):
+                with dbt_adapter.connection_named("create model"):
                     import pandas as pd
+
                     df = pd.read_csv(StringIO(csv))
                     dbt_adapter.execute(f"CREATE TABLE {schema}.{model_name} AS SELECT * FROM df")
             raw_code = sql if sql else csv
 
             if columns:
                 index = 1
-                table = CatalogTable(
-                    TableMetadata(type="BASE TABLE", schema=schema, name=model_name), {}, {})
+                table = CatalogTable(TableMetadata(type="BASE TABLE", schema=schema, name=model_name), {}, {})
                 catalog.nodes[unique_id] = table
                 for column, column_type in columns.items():
                     col_data = ColumnMetadata(type=column_type, index=index, name=column)
@@ -147,9 +157,7 @@ class DbtTestHelper:
                     "tags": ["test_tag"],
                 },
                 "tags": ["test_tag"],
-                "depends_on": {
-                    "nodes": depends_on
-                },
+                "depends_on": {"nodes": depends_on},
             }
             if patch_func:
                 patch_func(node_dict)
@@ -179,11 +187,12 @@ class DbtTestHelper:
             self.curr_manifest,
             self.base_manifest,
             self.base_catalog,
-            self.curr_catalog)
+            self.curr_catalog,
+        )
 
     def remove_model(self, model_name):
         dbt_adapter = self.adapter
-        with dbt_adapter.connection_named('cleanup'):
+        with dbt_adapter.connection_named("cleanup"):
             dbt_adapter.execute(f"DROP TABLE IF EXISTS {self.base_schema}.{model_name}")
             dbt_adapter.execute(f"DROP TABLE IF EXISTS {self.curr_schema}.{model_name} ")
 
@@ -219,15 +228,15 @@ class DbtTestHelper:
             if csv:
                 dbt_adapter = self.adapter
                 csv = textwrap.dedent(csv)
-                with dbt_adapter.connection_named('create source'):
+                with dbt_adapter.connection_named("create source"):
                     import pandas as pd
+
                     df = pd.read_csv(StringIO(csv))
                     dbt_adapter.execute(f"CREATE TABLE {schema}.{table_name} AS SELECT * FROM df")
 
             if columns:
                 index = 1
-                table = CatalogTable(
-                    TableMetadata(type="BASE TABLE", schema=schema, name=table_name), {}, {})
+                table = CatalogTable(TableMetadata(type="BASE TABLE", schema=schema, name=table_name), {}, {})
                 catalog.sources[unique_id] = table
                 for column, column_type in columns.items():
                     col_data = ColumnMetadata(type=column_type, index=index, name=column)
@@ -271,11 +280,12 @@ class DbtTestHelper:
             self.curr_manifest,
             self.base_manifest,
             self.base_catalog,
-            self.curr_catalog)
+            self.curr_catalog,
+        )
 
     def cleanup(self):
         dbt_adapter = self.adapter
-        with dbt_adapter.connection_named('cleanup'):
+        with dbt_adapter.connection_named("cleanup"):
             dbt_adapter.execute(f"DROP SCHEMA IF EXISTS {self.base_schema} CASCADE")
             dbt_adapter.execute(f"DROP SCHEMA IF EXISTS {self.curr_schema} CASCADE")
 

tests/adapter/dbt_adapter/test_dbt_adapter.py

@@ -1,15 +1,9 @@
 from recce.adapter.dbt_adapter import DbtAdapter, dbt_supported_registry
-from recce.models import RunType
 
 
 def test_dbt_adapter_support_tasks(dbt_test_helper):
     adapter: DbtAdapter = dbt_test_helper.context.adapter
 
-    # Test dbt package name
-    supported_dbt_packages = set([package.package_name for package in adapter.manifest.macros.values()])
-    assert 'audit_helper' in supported_dbt_packages
-    assert 'dbt_profiler' in supported_dbt_packages
-
     # Test dbt task support
     support_tasks = adapter.support_tasks()
 
@@ -29,12 +23,3 @@ def test_dbt_adapter_support_tasks_without_required_dbt_package(dbt_test_helper)
     for task_type in dbt_supported_registry:
         task = task_type.value
         assert task in support_tasks
-        if task == RunType.PROFILE_DIFF.value:
-            assert support_tasks[task] is False
-        elif task == RunType.VALUE_DIFF.value:
-            assert support_tasks[task] is False
-        elif task == RunType.VALUE_DIFF_DETAIL.value:
-            assert support_tasks[task] is False
-
-    # Check the query_diff_with_primary_key is not supported
-    assert support_tasks["query_diff_with_primary_key"] is False

tests/adapter/dbt_adapter/test_dbt_cll.py

@@ -3,25 +3,27 @@ from recce.adapter.dbt_adapter import DbtAdapter
 
 def test_cll_basic(dbt_test_helper):
     dbt_test_helper.create_model("model1", curr_sql="select 1 as c", curr_columns={"c": "int"})
-    dbt_test_helper.create_model("model2", curr_sql='select c from {{ ref("model1") }}', curr_columns={"c": "int"},
-                                 depends_on=["model1"])
+    dbt_test_helper.create_model(
+        "model2", curr_sql='select c from {{ ref("model1") }}', curr_columns={"c": "int"}, depends_on=["model1"]
+    )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
     result = adapter.get_cll_by_node_id("model1")
-    assert result['nodes']['model2']['columns']['c']['depends_on'][0].column == 'c'
-    assert result['nodes']['model2']['columns']['c']['depends_on'][0].node == 'model1'
+    assert result["nodes"]["model2"]["columns"]["c"]["depends_on"][0].column == "c"
+    assert result["nodes"]["model2"]["columns"]["c"]["depends_on"][0].node == "model1"
 
 
 def test_cll_table_alisa(dbt_test_helper):
     def patch_node(node):
-        node['alias'] = 'model1_alias'
+        node["alias"] = "model1_alias"
 
     dbt_test_helper.create_model("model1", curr_sql="select 1 as c", curr_columns={"c": "int"}, patch_func=patch_node)
-    dbt_test_helper.create_model("model2", curr_sql='select c from {{ ref("model1") }}', curr_columns={"c": "int"},
-                                 depends_on=["model1"])
+    dbt_test_helper.create_model(
+        "model2", curr_sql='select c from {{ ref("model1") }}', curr_columns={"c": "int"}, depends_on=["model1"]
+    )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
     result = adapter.get_cll_by_node_id("model1")
-    assert result['nodes']['model2']['columns']['c']['depends_on'][0].column == 'c'
-    assert result['nodes']['model2']['columns']['c']['depends_on'][0].node == 'model1'
+    assert result["nodes"]["model2"]["columns"]["c"]["depends_on"][0].column == "c"
+    assert result["nodes"]["model2"]["columns"]["c"]["depends_on"][0].node == "model1"
 
 
 def test_seed(dbt_test_helper):
@@ -32,20 +34,22 @@ def test_seed(dbt_test_helper):
     3,Charlie,35
     """
 
-    dbt_test_helper.create_model("seed1",
-                                 curr_csv=csv_data_curr,
-                                 curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"},
-                                 resource_type="seed")
+    dbt_test_helper.create_model(
+        "seed1",
+        curr_csv=csv_data_curr,
+        curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"},
+        resource_type="seed",
+    )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
     result = adapter.get_cll_by_node_id("seed1")
 
-    assert result['nodes']['seed1']['columns']['customer_id']['transformation_type'] == 'source'
-    assert len(result['nodes']['seed1']['columns']['customer_id']['depends_on']) == 0
+    assert result["nodes"]["seed1"]["columns"]["customer_id"]["transformation_type"] == "source"
+    assert len(result["nodes"]["seed1"]["columns"]["customer_id"]["depends_on"]) == 0
 
 
 def test_python_model(dbt_test_helper):
     def python_node(node):
-        node['language'] = 'python'
+        node["language"] = "python"
 
     csv_data_curr = """
     customer_id,name,age
@@ -53,20 +57,22 @@ def test_python_model(dbt_test_helper):
     2,Bob,25
     3,Charlie,35
     """
-    dbt_test_helper.create_model("model1",
-                                 curr_csv=csv_data_curr,
-                                 curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"})
-    dbt_test_helper.create_model("model2",
-                                 curr_csv=csv_data_curr,
-                                 curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"},
-                                 depends_on=["model1"],
-                                 patch_func=python_node)
+    dbt_test_helper.create_model(
+        "model1", curr_csv=csv_data_curr, curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"}
+    )
+    dbt_test_helper.create_model(
+        "model2",
+        curr_csv=csv_data_curr,
+        curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"},
+        depends_on=["model1"],
+        patch_func=python_node,
+    )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
-    assert not adapter.is_python_model('model1')
-    assert adapter.is_python_model('model2')
+    assert not adapter.is_python_model("model1")
+    assert adapter.is_python_model("model2")
 
     result = adapter.get_cll_by_node_id("model1")
-    assert result['nodes']['model2']['columns']['customer_id']['transformation_type'] == 'unknown'
+    assert result["nodes"]["model2"]["columns"]["customer_id"]["transformation_type"] == "unknown"
 
 
 def test_source(dbt_test_helper):
@@ -81,22 +87,23 @@ def test_source(dbt_test_helper):
         "source1",
         "table1",
         curr_csv=csv_data_curr,
-        curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"})
+        curr_columns={"customer_id": "varchar", "name": "varchar", "age": "int"},
+    )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
     result = adapter.get_cll_by_node_id("source1.table1")
-    assert result['nodes']['source1.table1']['columns']['customer_id']['transformation_type'] == 'source'
+    assert result["nodes"]["source1.table1"]["columns"]["customer_id"]["transformation_type"] == "source"
 
 
 def test_parse_error(dbt_test_helper):
     dbt_test_helper.create_model("model1", curr_sql="select 1 as c", curr_columns={"c": "int"})
-    dbt_test_helper.create_model("model2", curr_sql='this is not a valid sql', curr_columns={"c": "int"})
+    dbt_test_helper.create_model("model2", curr_sql="this is not a valid sql", curr_columns={"c": "int"})
     adapter: DbtAdapter = dbt_test_helper.context.adapter
     result = adapter.get_cll_by_node_id("model2")
-    assert result['nodes']['model2']['columns']['c']['transformation_type'] == 'unknown'
+    assert result["nodes"]["model2"]["columns"]["c"]["transformation_type"] == "unknown"
 
 
 def test_model_without_catalog(dbt_test_helper):
     dbt_test_helper.create_model("model1", curr_sql="select 1 as c")
     adapter: DbtAdapter = dbt_test_helper.context.adapter
     result = adapter.get_cll_by_node_id("model1")
-    assert not hasattr(result['nodes']['model1'], 'columns')
+    assert not hasattr(result["nodes"]["model1"], "columns")

tests/adapter/dbt_adapter/test_selector.py

@@ -21,13 +21,13 @@ def test_select(dbt_test_helper):
     adapter: DbtAdapter = dbt_test_helper.context.adapter
 
     # Test methods
-    node_ids = adapter.select_nodes('customers_1')
+    node_ids = adapter.select_nodes("customers_1")
     assert len(node_ids) == 1
-    node_ids = adapter.select_nodes('resource_type:model')
+    node_ids = adapter.select_nodes("resource_type:model")
     assert len(node_ids) == 2
-    node_ids = adapter.select_nodes('tag:test_tag')
+    node_ids = adapter.select_nodes("tag:test_tag")
     assert len(node_ids) == 2
-    node_ids = adapter.select_nodes('tag:test_tag2')
+    node_ids = adapter.select_nodes("tag:test_tag2")
     assert len(node_ids) == 0
     node_ids = adapter.select_nodes("config.materialized:incremental")
     assert len(node_ids) == 0
@@ -47,9 +47,9 @@ def test_select(dbt_test_helper):
     assert len(node_ids) == 2
     node_ids = adapter.select_nodes("config.materialized:table,tag:test_tag2")
     assert len(node_ids) == 0
-    node_ids = adapter.select_nodes(exclude='customers_1')
+    node_ids = adapter.select_nodes(exclude="customers_1")
     assert len(node_ids) == 1
-    node_ids = adapter.select_nodes('customers_1', exclude='customers_2')
+    node_ids = adapter.select_nodes("customers_1", exclude="customers_2")
     assert len(node_ids) == 1
 
     # Test graph operation
@@ -62,10 +62,10 @@ def test_select(dbt_test_helper):
     dbt_test_helper.create_snapshot("snapshot_1", csv_data_base, csv_data_curr)
     dbt_test_helper.create_model("use_snapshot", csv_data_base, csv_data_base, depends_on=["snapshot_1"])
 
-    node_ids = adapter.select_nodes('resource_type:snapshot')
+    node_ids = adapter.select_nodes("resource_type:snapshot")
     assert len(node_ids) == 1
 
-    node_ids = adapter.select_nodes('resource_type:snapshot+')
+    node_ids = adapter.select_nodes("resource_type:snapshot+")
     assert len(node_ids) == 2
 
     node_ids = adapter.select_nodes("state:modified,resource_type:snapshot")
@@ -172,30 +172,31 @@ def test_select_with_pacakage_mode_include_exclude(dbt_test_helper):
     Only customers_3 is changed
     """
     dbt_test_helper.create_model("customers_1", csv_data_base, csv_data_base)
-    dbt_test_helper.create_model("customers_2", csv_data_base, csv_data_base, depends_on=['customers_1'])
-    dbt_test_helper.create_model("customers_3", csv_data_base, csv_data_curr, depends_on=['customers_2'])
-    dbt_test_helper.create_model("customers_4", csv_data_base, csv_data_base, depends_on=['customers_3'],
-                                 package_name='other_package')
-    dbt_test_helper.create_model("customers_5", csv_data_base, csv_data_base, depends_on=['customers_3'])
+    dbt_test_helper.create_model("customers_2", csv_data_base, csv_data_base, depends_on=["customers_1"])
+    dbt_test_helper.create_model("customers_3", csv_data_base, csv_data_curr, depends_on=["customers_2"])
+    dbt_test_helper.create_model(
+        "customers_4", csv_data_base, csv_data_base, depends_on=["customers_3"], package_name="other_package"
+    )
+    dbt_test_helper.create_model("customers_5", csv_data_base, csv_data_base, depends_on=["customers_3"])
 
     adapter: DbtAdapter = dbt_test_helper.context.adapter
 
-    node_ids = adapter.select_nodes(packages=['other_package'])
+    node_ids = adapter.select_nodes(packages=["other_package"])
     assert len(node_ids) == 1
 
-    node_ids = adapter.select_nodes(view_mode='changed_models')
+    node_ids = adapter.select_nodes(view_mode="changed_models")
     assert len(node_ids) == 4
 
-    node_ids = adapter.select_nodes(view_mode='changed_models', packages=['other_package'])
+    node_ids = adapter.select_nodes(view_mode="changed_models", packages=["other_package"])
     assert len(node_ids) == 1
 
-    node_ids = adapter.select_nodes(view_mode='changed_models', packages=['other_package'], exclude='customers_4')
+    node_ids = adapter.select_nodes(view_mode="changed_models", packages=["other_package"], exclude="customers_4")
     assert len(node_ids) == 0
-    node_ids = adapter.select_nodes(view_mode='changed_models', packages=['other_package'], select='customers_1+')
+    node_ids = adapter.select_nodes(view_mode="changed_models", packages=["other_package"], select="customers_1+")
     assert len(node_ids) == 1
-    node_ids = adapter.select_nodes(view_mode='changed_models', select='+customers_5')
+    node_ids = adapter.select_nodes(view_mode="changed_models", select="+customers_5")
     assert len(node_ids) == 3
-    node_ids = adapter.select_nodes(view_mode='all', select='+customers_5')
+    node_ids = adapter.select_nodes(view_mode="all", select="+customers_5")
     assert len(node_ids) == 4
-    node_ids = adapter.select_nodes(select='+customers_5')
+    node_ids = adapter.select_nodes(select="+customers_5")
     assert len(node_ids) == 4

tests/tasks/test_histogram.py

@@ -1,6 +1,10 @@
 import pytest
 
-from recce.tasks.histogram import HistogramDiffTask, HistogramDiffCheckValidator, _is_histogram_supported
+from recce.tasks.histogram import (
+    HistogramDiffCheckValidator,
+    HistogramDiffTask,
+    _is_histogram_supported,
+)
 
 
 def test_histogram(dbt_test_helper):
@@ -14,11 +18,7 @@ def test_histogram(dbt_test_helper):
 
     dbt_test_helper.create_model("customers", csv_data, csv_data)
 
-    params = {
-        "model": "customers",
-        "column_name": "age",
-        "column_type": "int"
-    }
+    params = {"model": "customers", "column_name": "age", "column_type": "int"}
 
     task = HistogramDiffTask(params)
     run_result = task.execute()
@@ -30,13 +30,13 @@ def test_histogram(dbt_test_helper):
     #     'bin_edges': [25, 26, ..., 51],
     #     'labels': ['25-26', ..., '51-52']
     # }
-    assert run_result['current']['counts'][0] == 1
-    assert run_result['current']['counts'][-1] == 1
-    assert run_result['current']['total'] == 4
-    assert run_result['min'] == 25
-    assert run_result['max'] == 50
-    assert run_result['bin_edges'][0] == 25
-    assert run_result['bin_edges'][-1] == 51
+    assert run_result["current"]["counts"][0] == 1
+    assert run_result["current"]["counts"][-1] == 1
+    assert run_result["current"]["total"] == 4
+    assert run_result["min"] == 25
+    assert run_result["max"] == 50
+    assert run_result["bin_edges"][0] == 25
+    assert run_result["bin_edges"][-1] == 51
 
 
 def test_histogram_emtpy(dbt_test_helper):
@@ -56,74 +56,66 @@ def test_histogram_emtpy(dbt_test_helper):
     dbt_test_helper.create_model("customers2", csv_data, csv_data_zero)
     dbt_test_helper.create_model("customers3", csv_data_zero, csv_data)
 
-    params = {
-        "model": "customers",
-        "column_name": "age",
-        "column_type": "int"
-    }
+    params = {"model": "customers", "column_name": "age", "column_type": "int"}
 
     task = HistogramDiffTask(params)
     run_result = task.execute()
 
-    assert len(run_result['current']['counts']) == 0
-    assert run_result['current']['total'] == 0
-    assert run_result['min'] is None
-    assert run_result['max'] is None
-    assert len(run_result['bin_edges']) == 0
+    assert len(run_result["current"]["counts"]) == 0
+    assert run_result["current"]["total"] == 0
+    assert run_result["min"] is None
+    assert run_result["max"] is None
+    assert len(run_result["bin_edges"]) == 0
 
-    params = {
-        "model": "customers2",
-        "column_name": "age",
-        "column_type": "int"
-    }
+    params = {"model": "customers2", "column_name": "age", "column_type": "int"}
 
     task = HistogramDiffTask(params)
     run_result = task.execute()
-    assert run_result['base']['counts'][0] == 1
-    assert run_result['base']['counts'][-1] == 1
-    assert run_result['base']['total'] == 4
-    assert run_result['current']['counts'][0] == 0
-    assert run_result['current']['counts'][-1] == 0
-    assert run_result['current']['total'] == 0
-    assert run_result['min'] == 25
-    assert run_result['max'] == 50
-    assert run_result['bin_edges'][0] == 25
-    assert run_result['bin_edges'][-1] == 51
-
-    params = {
-        "model": "customers3",
-        "column_name": "age",
-        "column_type": "int"
-    }
+    assert run_result["base"]["counts"][0] == 1
+    assert run_result["base"]["counts"][-1] == 1
+    assert run_result["base"]["total"] == 4
+    assert run_result["current"]["counts"][0] == 0
+    assert run_result["current"]["counts"][-1] == 0
+    assert run_result["current"]["total"] == 0
+    assert run_result["min"] == 25
+    assert run_result["max"] == 50
+    assert run_result["bin_edges"][0] == 25
+    assert run_result["bin_edges"][-1] == 51
+
+    params = {"model": "customers3", "column_name": "age", "column_type": "int"}
 
     task = HistogramDiffTask(params)
     run_result = task.execute()
-    assert run_result['base']['counts'][0] == 0
-    assert run_result['base']['counts'][-1] == 0
-    assert run_result['base']['total'] == 0
-    assert run_result['current']['counts'][0] == 1
-    assert run_result['current']['counts'][-1] == 1
-    assert run_result['current']['total'] == 4
-    assert run_result['min'] == 25
-    assert run_result['max'] == 50
-    assert run_result['bin_edges'][0] == 25
-    assert run_result['bin_edges'][-1] == 51
+    assert run_result["base"]["counts"][0] == 0
+    assert run_result["base"]["counts"][-1] == 0
+    assert run_result["base"]["total"] == 0
+    assert run_result["current"]["counts"][0] == 1
+    assert run_result["current"]["counts"][-1] == 1
+    assert run_result["current"]["total"] == 4
+    assert run_result["min"] == 25
+    assert run_result["max"] == 50
+    assert run_result["bin_edges"][0] == 25
+    assert run_result["bin_edges"][-1] == 51
 
 
 def test_validator():
     def validate(params: dict = {}, view_options: dict = {}):
-        HistogramDiffCheckValidator().validate({
-            'name': 'test',
-            'type': 'histogram_diff',
-            'params': params,
-            'view_options': view_options,
-        })
-
-    validate({
-        "model": "customers",
-        "column_name": "age",
-        "column_type": "int",
-    })
+        HistogramDiffCheckValidator().validate(
+            {
+                "name": "test",
+                "type": "histogram_diff",
+                "params": params,
+                "view_options": view_options,
+            }
+        )
+
+    validate(
+        {
+            "model": "customers",
+            "column_name": "age",
+            "column_type": "int",
+        }
+    )
 
     with pytest.raises(ValueError):
         validate({})

tests/tasks/test_lineage.py

@@ -3,40 +3,53 @@ import pytest
 
 def test_validator():
     from recce.tasks.lineage import LineageDiffCheckValidator
+
     validator = LineageDiffCheckValidator()
 
     def validate(params: dict):
-        validator.validate({
-            'name': 'test',
-            'type': 'schema_diff',
-            'params': params,
-        })
+        validator.validate(
+            {
+                "name": "test",
+                "type": "schema_diff",
+                "params": params,
+            }
+        )
 
     # Select all models
     validate({})
 
     # Select by selector
-    validate({
-        'select': 'customers',
-        'exclude': 'customers',
-        'packages': ['jaffle_shop'],
-        'view_mode': 'all',
-    })
+    validate(
+        {
+            "select": "customers",
+            "exclude": "customers",
+            "packages": ["jaffle_shop"],
+            "view_mode": "all",
+        }
+    )
 
     # packages should be an array
     with pytest.raises(ValueError):
-        validate({
-            'packages': 'jaffle_shop',
-        })
+        validate(
+            {
+                "packages": "jaffle_shop",
+            }
+        )
 
     # view_mode should be 'all' or 'changed_models'
-    validate({
-        'view_mode': None,
-    })
-    validate({
-        'view_mode': 'all',
-    })
+    validate(
+        {
+            "view_mode": None,
+        }
+    )
+    validate(
+        {
+            "view_mode": "all",
+        }
+    )
     with pytest.raises(ValueError):
-        validate({
-            'view_mode': 'abc',
-        })
+        validate(
+            {
+                "view_mode": "abc",
+            }
+        )