pyneoinstance 4.0.3__tar.gz → 4.0.5__tar.gz

{pyneoinstance-4.0.3/pyneoinstance.egg-info → pyneoinstance-4.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyneoinstance
-Version: 4.0.3
+Version: 4.0.5
 Summary: Handle Neo4j read and write transactions and provide methods for performing graph EDA
 Author-email: "Jose A. Alvarado-Guzman" <jose.alvarado-guzman@neo4j.com>
 License: GNU GENERAL PUBLIC LICENSE
@@ -686,6 +686,9 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
 Classifier: Development Status :: 5 - Production/Stable
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -695,6 +698,8 @@ Requires-Dist: pandas>=1.3.0
 Requires-Dist: PyYAML>=5.1
 Requires-Dist: pyvis>=0.3.0
 Requires-Dist: neo4j-viz>=1.0.0
+Requires-Dist: comm>=0.1.3
+Requires-Dist: traitlets>=5.10.0
 Provides-Extra: docs
 Requires-Dist: mkdocs-material>=9.0.0; extra == "docs"
 Requires-Dist: mkdocstrings[python]>=0.24.0; extra == "docs"
@@ -965,42 +970,42 @@ The get_node_label_freq method provides the frequency and relative frequency of
 ```python
 graph.get_node_label_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/nodefreq.png" width="250">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/nodefreq.png" width="250">
 
 ### Relationship Type Frequency
 The get_rela_type_freq method provides the frequency and relative frequency of all relationship types in the graph.
 ```python
 graph.get_rela_type_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/relaFreq.png" width="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/relaFreq.png" width="300">
 
 ### Relationship Source-Target Frequency
 The get_rela_source_target_freq method provides the frequency and relative frequency of all relationship types considering the node label of the source and targets of the relationships.
 ```python
 graph.get_rela_type_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/relaSourceTargetFreq.png" width="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/relaSourceTargetFreq.png" width="300">
 
 ### Properties
 The get_properties method provides information about all the nodes and relationships properties.
 ```python
 graph.get_properties(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/properties.png" width="350" hight="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/properties.png" width="350" hight="300">
 
 ### Constraints
 The get_constraints method provides information about all the graph constraints.
 ```python
 graph.get_constraints(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/constraints.png" width="350" hight="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/constraints.png" width="350" hight="300">
 
 ### Indexes
 The get_indexes method provides information about all the graph indexes.
 ```python
 graph.get_indexes(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/indexes.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/indexes.png" width="450" hight="400">
 
 ### Query Graph Visualization
 The `get_query_visualization` method executes a Cypher query that returns nodes and relationships and renders the result as an interactive graph. Node labels are displayed as captions and all properties appear as tooltips on hover.
@@ -1032,8 +1037,8 @@ schema.write_html('schema.html')
 
 Open `schema.html` in your browser to explore the schema interactively.
 
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/schema.png" width="450" hight="400">
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/schemaHtml.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/schema.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/schemaHtml.png" width="450" hight="400">
 
 ## More detailed examples
 

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/README.md

@@ -263,42 +263,42 @@ The get_node_label_freq method provides the frequency and relative frequency of
 ```python
 graph.get_node_label_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/nodefreq.png" width="250">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/nodefreq.png" width="250">
 
 ### Relationship Type Frequency
 The get_rela_type_freq method provides the frequency and relative frequency of all relationship types in the graph.
 ```python
 graph.get_rela_type_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/relaFreq.png" width="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/relaFreq.png" width="300">
 
 ### Relationship Source-Target Frequency
 The get_rela_source_target_freq method provides the frequency and relative frequency of all relationship types considering the node label of the source and targets of the relationships.
 ```python
 graph.get_rela_type_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/relaSourceTargetFreq.png" width="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/relaSourceTargetFreq.png" width="300">
 
 ### Properties
 The get_properties method provides information about all the nodes and relationships properties.
 ```python
 graph.get_properties(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/properties.png" width="350" hight="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/properties.png" width="350" hight="300">
 
 ### Constraints
 The get_constraints method provides information about all the graph constraints.
 ```python
 graph.get_constraints(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/constraints.png" width="350" hight="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/constraints.png" width="350" hight="300">
 
 ### Indexes
 The get_indexes method provides information about all the graph indexes.
 ```python
 graph.get_indexes(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/indexes.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/indexes.png" width="450" hight="400">
 
 ### Query Graph Visualization
 The `get_query_visualization` method executes a Cypher query that returns nodes and relationships and renders the result as an interactive graph. Node labels are displayed as captions and all properties appear as tooltips on hover.
@@ -330,8 +330,8 @@ schema.write_html('schema.html')
 
 Open `schema.html` in your browser to explore the schema interactively.
 
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/schema.png" width="450" hight="400">
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/schemaHtml.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/schema.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/schemaHtml.png" width="450" hight="400">
 
 ## More detailed examples
 

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/pyneoinstance/__init__.py

@@ -1,5 +1,5 @@
 from .database.neo4jdbms import Neo4jInstance
 from .fileload.loadyaml import load_yaml_file
 
-__version__ = '4.0.3'
+__version__ = '4.0.5'
 __all__ = ['Neo4jInstance', 'load_yaml_file']

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/pyneoinstance/database/neo4jdbms.py

@@ -15,12 +15,15 @@ from concurrent.futures import ThreadPoolExecutor
 import logging
 import os
 import re
+import time
 from pandas import DataFrame
 from neo4j import GraphDatabase
 from pyvis.network import Network
 from neo4j_viz import VisualizationGraph
 from neo4j_viz.neo4j import from_neo4j
 from neo4j.exceptions import ServiceUnavailable
+from neo4j.exceptions import SessionExpired
+from neo4j.exceptions import TransientError
 from neo4j.exceptions import AuthError
 from neo4j.exceptions import ClientError
 from neo4j.exceptions import ConfigurationError
@@ -34,6 +37,53 @@ _IMPLICIT_TX_RE = re.compile(
     re.IGNORECASE,
 )
 
+# Errors treated as transient/retryable for auto-commit queries (run via
+# session.run). session.execute_read/execute_write retry these automatically
+# through the driver's managed-transaction logic; session.run does not.
+_RETRYABLE_EXCEPTIONS = (ServiceUnavailable, SessionExpired, TransientError)
+
+
+def _run_autocommit_with_retry(session, consume, query, max_retries=5,
+                                initial_delay=1.0, backoff_factor=2.0, **kwargs):
+    """Run an auto-commit query (via session.run) with manual retry.
+
+    Queries using ``CALL { ... } IN TRANSACTIONS`` or
+    ``IN CONCURRENT TRANSACTIONS`` are only valid in an implicit
+    (auto-commit) transaction, so they can't be executed through
+    session.execute_read/execute_write and don't benefit from the driver's
+    built-in managed-transaction retry behavior. This re-runs the query and
+    re-consumes its result from scratch on transient failures such as
+    ServiceUnavailable.
+
+    Parameters
+    ----------
+    session : neo4j.Session
+        The session to run the query on.
+    consume : Callable[[neo4j.Result], Any]
+        Function that consumes the Result object (e.g. reads values, builds
+        a graph, reads counters) and returns the desired output.
+    query : str
+        The Cypher query to run.
+    max_retries : int, optional
+        Maximum number of retry attempts after the initial try.
+    initial_delay : float, optional
+        Delay in seconds before the first retry.
+    backoff_factor : float, optional
+        Multiplier applied to the delay after each subsequent retry.
+    kwargs
+        Cypher query parameters.
+    """
+    delay = initial_delay
+    for attempt in range(max_retries + 1):
+        try:
+            result = session.run(query, **kwargs)
+            return consume(result)
+        except _RETRYABLE_EXCEPTIONS:
+            if attempt == max_retries:
+                raise
+            time.sleep(delay)
+            delay *= backoff_factor
+
 def _get_driver(neo_info : Dict[str, str]):
     try:
         if neo_info['encrypted']:
@@ -204,8 +254,11 @@ class Neo4jInstance:
                 if _IMPLICIT_TX_RE.search(query):
                     # 'CALL { ... } IN TRANSACTIONS' is only valid in an implicit
                     # (auto-commit) transaction, so it can't go through execute_read.
-                    read_result = session.run(query, **params)
-                    result = DataFrame(read_result.values(), columns=read_result.keys())
+                    # session.run isn't retried by the driver, so retry manually.
+                    result = _run_autocommit_with_retry(
+                        session,
+                        lambda r: DataFrame(r.values(), columns=r.keys()),
+                        query, **params)
                 else:
                     result = session.execute_read(
                         _read_transaction_function,query=query,**params)
@@ -256,9 +309,12 @@ class Neo4jInstance:
                 if _IMPLICIT_TX_RE.search(query):
                     # 'CALL { ... } IN TRANSACTIONS' is only valid in an implicit
                     # (auto-commit) transaction, so it can't go through execute_read.
-                    graph_result = session.run(query, **params)
-                    list(graph_result)  # consume all records so .graph() is populated
-                    graph = graph_result.graph()
+                    # session.run isn't retried by the driver, so retry manually.
+                    def _consume_graph(result):
+                        list(result)  # consume records so .graph() is populated
+                        return result.graph()
+                    graph = _run_autocommit_with_retry(
+                        session, _consume_graph, query, **params)
                 else:
                     graph = session.execute_read(
                         _graph_transaction_function, query=query, **params)
@@ -695,8 +751,8 @@ class Neo4jInstance:
                     'targetLabel': nodes[1], 'frequency': value}
             stats_dicts.append(info)
         stats_df = DataFrame(stats_dicts)
-        stats_df.sort_values(by=['relationshipType','sourceLabel','targetLabel'],inplace=True)
-        stats_df.reset_index(drop=True, inplace=True)
+        stats_df = stats_df.sort_values(by=['relationshipType','sourceLabel','targetLabel'])
+        stats_df = stats_df.reset_index(drop=True)
         return stats_df
 
     def get_schema_visualization(self, database: Optional[str] = None) -> Network:
@@ -775,7 +831,10 @@ class Neo4jInstance:
             if _IMPLICIT_TX_RE.search(query):
                 # 'CALL { ... } IN TRANSACTIONS' is only valid in an implicit
                 # (auto-commit) transaction, so it can't go through execute_write.
-                results = session.run(query, **kwargs).consume().counters.__dict__
+                # session.run isn't retried by the driver, so retry manually.
+                results = _run_autocommit_with_retry(
+                    session, lambda r: r.consume().counters.__dict__,
+                    query, **kwargs)
             else:
                 results = session.execute_write(
                     _write_transaction_function, query,

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5/pyneoinstance.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyneoinstance
-Version: 4.0.3
+Version: 4.0.5
 Summary: Handle Neo4j read and write transactions and provide methods for performing graph EDA
 Author-email: "Jose A. Alvarado-Guzman" <jose.alvarado-guzman@neo4j.com>
 License: GNU GENERAL PUBLIC LICENSE
@@ -686,6 +686,9 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
 Classifier: Development Status :: 5 - Production/Stable
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -695,6 +698,8 @@ Requires-Dist: pandas>=1.3.0
 Requires-Dist: PyYAML>=5.1
 Requires-Dist: pyvis>=0.3.0
 Requires-Dist: neo4j-viz>=1.0.0
+Requires-Dist: comm>=0.1.3
+Requires-Dist: traitlets>=5.10.0
 Provides-Extra: docs
 Requires-Dist: mkdocs-material>=9.0.0; extra == "docs"
 Requires-Dist: mkdocstrings[python]>=0.24.0; extra == "docs"
@@ -965,42 +970,42 @@ The get_node_label_freq method provides the frequency and relative frequency of
 ```python
 graph.get_node_label_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/nodefreq.png" width="250">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/nodefreq.png" width="250">
 
 ### Relationship Type Frequency
 The get_rela_type_freq method provides the frequency and relative frequency of all relationship types in the graph.
 ```python
 graph.get_rela_type_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/relaFreq.png" width="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/relaFreq.png" width="300">
 
 ### Relationship Source-Target Frequency
 The get_rela_source_target_freq method provides the frequency and relative frequency of all relationship types considering the node label of the source and targets of the relationships.
 ```python
 graph.get_rela_type_freq(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/relaSourceTargetFreq.png" width="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/relaSourceTargetFreq.png" width="300">
 
 ### Properties
 The get_properties method provides information about all the nodes and relationships properties.
 ```python
 graph.get_properties(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/properties.png" width="350" hight="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/properties.png" width="350" hight="300">
 
 ### Constraints
 The get_constraints method provides information about all the graph constraints.
 ```python
 graph.get_constraints(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/constraints.png" width="350" hight="300">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/constraints.png" width="350" hight="300">
 
 ### Indexes
 The get_indexes method provides information about all the graph indexes.
 ```python
 graph.get_indexes(database='mydbname')
 ```
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/indexes.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/indexes.png" width="450" hight="400">
 
 ### Query Graph Visualization
 The `get_query_visualization` method executes a Cypher query that returns nodes and relationships and renders the result as an interactive graph. Node labels are displayed as captions and all properties appear as tooltips on hover.
@@ -1032,8 +1037,8 @@ schema.write_html('schema.html')
 
 Open `schema.html` in your browser to explore the schema interactively.
 
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/schema.png" width="450" hight="400">
-<img src="https://s3.amazonaws.com/neo4j-ps-ds-bootcamp/img/schemaHtml.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/schema.png" width="450" hight="400">
+<img src="https://raw.githubusercontent.com/jose-alvarado-guzman/pyneoingest/main/images/schemaHtml.png" width="450" hight="400">
 
 ## More detailed examples
 

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/pyneoinstance.egg-info/SOURCES.txt

@@ -18,7 +18,6 @@ pyneoinstance/fileload/helpers.py
 pyneoinstance/fileload/loadyaml.py
 pyneoinstance/util/__init__.py
 pyneoinstance/util/functions.py
-test/age_data.csv
 test/cypher_queries.yaml
 test/load_files.yaml
 test/movies.csv

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/pyneoinstance.egg-info/requires.txt

@@ -3,6 +3,8 @@ pandas>=1.3.0
 PyYAML>=5.1
 pyvis>=0.3.0
 neo4j-viz>=1.0.0
+comm>=0.1.3
+traitlets>=5.10.0
 
 [docs]
 mkdocs-material>=9.0.0

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/pyneoinstance.egg-info/top_level.txt

@@ -2,4 +2,3 @@ dist
 docs
 images
 pyneoinstance
-site

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyneoinstance"
-version = "4.0.3"
+version = "4.0.5"
 description = "Handle Neo4j read and write transactions and provide methods for performing graph EDA"
 readme = "README.md"
 authors = [{ name = "Jose A. Alvarado-Guzman",email = "jose.alvarado-guzman@neo4j.com" }]
@@ -17,6 +17,9 @@ classifiers = [
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Programming Language :: Python :: 3.15",
     "Development Status :: 5 - Production/Stable",
 ]
 keywords = ["neo4j", "graph", "cypher"]
@@ -25,7 +28,9 @@ dependencies = [
     "pandas>=1.3.0",
     "PyYAML>=5.1",
     "pyvis>=0.3.0",
-    "neo4j-viz>=1.0.0"
+    "neo4j-viz>=1.0.0",
+    "comm>=0.1.3",
+    "traitlets>=5.10.0"
 ]
 requires-python = ">=3.10"
 

{pyneoinstance-4.0.3 → pyneoinstance-4.0.5}/test/test_neo4jdb.py

@@ -10,10 +10,14 @@
 """
 import os
 import unittest
+from contextlib import contextmanager
+from unittest.mock import patch
+import neo4j
 import pandas as pd
 import numpy as np
 import pandas.testing as pd_testing
 from neo4j.exceptions import ClientError
+from neo4j.exceptions import ServiceUnavailable
 from neo4j_viz import VisualizationGraph
 from .context import database
 from .context import fileload
@@ -40,6 +44,32 @@ def _clean_db(graph, db):
         ["CALL apoc.schema.assert({},{})", "MATCH(n) DETACH DELETE n"], db)
 
 
+@contextmanager
+def _patch_flaky_session_run(fail_times=1):
+    """Patch neo4j.Session.run so the first `fail_times` calls raise a
+    transient ServiceUnavailable, then delegate to the real driver.
+
+    Used to verify that 'IN TRANSACTIONS'/'IN CONCURRENT TRANSACTIONS'
+    queries (which go through session.run rather than execute_read/
+    execute_write, and so don't get the driver's built-in managed-
+    transaction retries) are retried by the manual retry wrapper instead
+    of failing outright. Also patches time.sleep so the retry backoff
+    doesn't slow the test down.
+    """
+    original_run = neo4j.Session.run
+    call_count = {'n': 0}
+
+    def flaky_run(session_self, *args, **kwargs):
+        call_count['n'] += 1
+        if call_count['n'] <= fail_times:
+            raise ServiceUnavailable('simulated transient failure')
+        return original_run(session_self, *args, **kwargs)
+
+    with patch.object(neo4j.Session, 'run', flaky_run), \
+         patch.object(database.neo4jdbms.time, 'sleep', return_value=None):
+        yield call_count
+
+
 class TestWriteMethods(unittest.TestCase):
     """Test write operations. Each test gets a clean database."""
 
@@ -125,6 +155,24 @@ class TestWriteMethods(unittest.TestCase):
         self.assertEqual(
             {'nodes_created': 3, 'labels_added': 3, 'properties_set': 3}, result)
 
+    def test_execute_write_query_in_transactions_retries_on_transient_failure(self):
+        """An 'IN TRANSACTIONS' write query is retried and still succeeds
+        after a transient ServiceUnavailable failure, since it goes through
+        session.run (not execute_write) and isn't covered by the driver's
+        built-in managed-transaction retries."""
+        query = """
+            UNWIND range(1, 3) AS i
+            CALL (i) {
+                CREATE (:Actor {id: i})
+            } IN TRANSACTIONS OF 1 ROWS
+        """
+        with _patch_flaky_session_run(fail_times=1) as call_count:
+            result = self.graph.execute_write_query(query, self.db)
+
+        self.assertEqual(call_count['n'], 2)
+        self.assertEqual(
+            {'nodes_created': 3, 'labels_added': 3, 'properties_set': 3}, result)
+
 
 class TestReadAndEdaMethods(unittest.TestCase):
     """Test read and EDA methods. Data is loaded once for the entire class."""
@@ -203,6 +251,26 @@ class TestReadAndEdaMethods(unittest.TestCase):
         result = self.graph.execute_read_query(query, self.db)
         self.assertEqual(plain_count, result['row_num'][0])
 
+    def test_execute_read_query_in_transactions_retries_on_transient_failure(self):
+        """An 'IN TRANSACTIONS' read query is retried and still succeeds
+        after a transient ServiceUnavailable failure, since it goes through
+        session.run (not execute_read) and isn't covered by the driver's
+        built-in managed-transaction retries."""
+        plain_count = self.graph.execute_read_query(
+            "MATCH (p:Person) RETURN count(p) AS row_num", self.db)['row_num'][0]
+        query = """
+            MATCH (p:Person)
+            CALL (p) {
+                RETURN p AS person
+            } IN TRANSACTIONS OF 1000 ROWS
+            RETURN count(person) AS row_num
+        """
+        with _patch_flaky_session_run(fail_times=1) as call_count:
+            result = self.graph.execute_read_query(query, self.db)
+
+        self.assertEqual(call_count['n'], 2)
+        self.assertEqual(plain_count, result['row_num'][0])
+
     def test_get_node_label_freq(self):
         """Test node label frequency EDA."""
         result = self.graph.get_node_label_freq(self.db)