fix test

nakagami · Feb 4, 2024 · 6fc4dfa · 6fc4dfa
1 parent 70789fa
commit 6fc4dfa
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/test_db2.py b/test_db2.py
@@ -142,7 +142,7 @@ def test_error(self):
             cur.execute("invalid query"),
 
     @unittest.skip
-    def test_issue_18(self):
+    def test_issue18(self):
         cur = self.connection.cursor()
         try:
             cur.execute("DROP TABLE test_issue18")
@@ -158,7 +158,10 @@ def test_issue_18(self):
         for _ in range(count):
             cur.execute("INSERT INTO test_issue18(s) values(?)", [s])
         cur.execute("SELECT * FROM test_issue18")
-        self.assertEqual(cur.fetchall(), [s] * count)
+        self.assertEqual(
+            list(cur.fetchall()),
+            [(s,) for _ in range(count)],
+        )
 
 
 class TestDataType(unittest.TestCase):

diff --git a/test_derby.py b/test_derby.py
@@ -270,7 +270,7 @@ def test_double(self):
         self.assertEqual(cur.fetchall(), [(-1, -1, -1.0, -1.0)])
 
     @unittest.skip
-    def test_issue_18(self):
+    def test_issue18(self):
         cur = self.connection.cursor()
         try:
             cur.execute("DROP TABLE test_issue18")
@@ -281,12 +281,15 @@ def test_issue_18(self):
                 s varchar(4096)
             )
         """)
+        s = "x" * 4096
         count = 20
         for _ in range(count):
-            s = "x" * 4096
             cur.execute(f"INSERT INTO test_issue18(s) values('{s}')")
         cur.execute("SELECT * FROM test_issue18")
-        self.assertEqual(cur.fetchall(), [s] * count)
+        self.assertEqual(
+            list(cur.fetchall()),
+            [(s,) for _ in range(count)],
+        )
 
     def tearDown(self):
         self.connection.close()