Fix format

Author: zprobst

nodestream_plugin_neo4j/database_connector.py

@@ -60,7 +60,9 @@ def __init__(
         self.set_first_ingested_at = set_first_ingested_at
 
     def make_query_executor(self) -> QueryExecutor:
-        query_builder = Neo4jIngestQueryBuilder(self.use_apoc, self.set_first_ingested_at)
+        query_builder = Neo4jIngestQueryBuilder(
+            self.use_apoc, self.set_first_ingested_at
+        )
         return Neo4jQueryExecutor(
             self.database_connection,
             query_builder,

tests/e2e/test_neo4j_pipelines.py

@@ -34,7 +34,9 @@ def validate_timestamps(session):
         """
     )
 
-    assert nodes_with_first_ingested_at.single()["count"] == total_nodes.single()["count"]
+    assert (
+        nodes_with_first_ingested_at.single()["count"] == total_nodes.single()["count"]
+    )
 
     # Check that all relationships have a last_ingested_at property
     relationships_with_last_ingested_at = session.run(
@@ -51,7 +53,10 @@ def validate_timestamps(session):
         """
     )
 
-    assert relationships_with_last_ingested_at.single()["count"] == total_relationships.single()["count"]
+    assert (
+        relationships_with_last_ingested_at.single()["count"]
+        == total_relationships.single()["count"]
+    )
 
 
 def validate_airports(session):
@@ -162,7 +167,11 @@ def validate_ttl_seperation_between_node_object_types(session):
 PIPELINE_TESTS = [
     ("airports", [validate_airports, valiudate_airport_country], "my-neo4j-db"),
     ("fifa", [validate_fifa_player_count, validate_fifa_mo_club], "my-neo4j-db"),
-    ("airpoints", [validate_airports, valiudate_airport_country, validate_timestamps], "my-creation-ts-db"),
+    (
+        "airpoints",
+        [validate_airports, valiudate_airport_country, validate_timestamps],
+        "my-creation-ts-db",
+    ),
 ]
 
 TTL_TESTS = [

tests/unit/test_ingest_query_builder.py

@@ -279,9 +279,21 @@ def test_node_update_generates_expected_queries(
 @pytest.mark.parametrize(
     "node,expected_query,node_creation_rule",
     [
-        [SIMPLE_NODE, SIMPLE_NODE_EXPECTED_QUERY_WITH_CREATED_TIMESTAMP, NodeCreationRule.EAGER],
-        [COMPLEX_NODE, COMPLEX_NODE_EXPECTED_QUERY_WITH_CREATED_TIMESTAMP, NodeCreationRule.EAGER],
-        [COMPLEX_NODE_TWO, COMPLEX_NODE_TWO_EXPECTED_QUERY_WITH_CREATED_TIMESTAMP, NodeCreationRule.EAGER],
+        [
+            SIMPLE_NODE,
+            SIMPLE_NODE_EXPECTED_QUERY_WITH_CREATED_TIMESTAMP,
+            NodeCreationRule.EAGER,
+        ],
+        [
+            COMPLEX_NODE,
+            COMPLEX_NODE_EXPECTED_QUERY_WITH_CREATED_TIMESTAMP,
+            NodeCreationRule.EAGER,
+        ],
+        [
+            COMPLEX_NODE_TWO,
+            COMPLEX_NODE_TWO_EXPECTED_QUERY_WITH_CREATED_TIMESTAMP,
+            NodeCreationRule.EAGER,
+        ],
         [SIMPLE_NODE, SIMPLE_NODE_EXPECTED_QUERY_ON_MATCH, NodeCreationRule.MATCH_ONLY],
     ],
 )