Add error assertion for Python (Result.single) (#373)

neo4j-drivers · Jan 20, 2022 · 6b918ba · 6b918ba
1 parent acf0ed2
commit 6b918ba
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/stub/iteration/test_result_single.py b/tests/stub/iteration/test_result_single.py
@@ -15,8 +15,10 @@ def _assert_not_exactly_one_record_error(self, error):
         self.assertIsInstance(error, types.DriverError)
         driver = get_driver_name()
         if driver in ["python"]:
-            self.assertEqual("<class 'ToBeDecided'>",
-                             error.errorType)
+            self.assertEqual(
+                "<class 'neo4j.exceptions.ResultNotSingleError'>",
+                error.errorType
+            )
         elif driver in ["ruby"]:
             self.assertEqual(
                 "Neo4j::Driver::Exceptions::NoSuchRecordException",