From 6b918ba277ee89624722ead3f514fdbc2ec26014 Mon Sep 17 00:00:00 2001
From: Robsdedude <rouven.bauer@neotechnology.com>
Date: Thu, 20 Jan 2022 20:04:34 +0100
Subject: [PATCH] Add error assertion for Python (`Result.single`) (#373)

---
 tests/stub/iteration/test_result_single.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/stub/iteration/test_result_single.py b/tests/stub/iteration/test_result_single.py
index 58e10df2b..c1d71c547 100644
--- 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",