fix test

nakagami · Aug 23, 2024 · b8b6a86 · b8b6a86
1 parent bc2799d
commit b8b6a86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cymysql/tests/test_basic.py b/cymysql/tests/test_basic.py
@@ -100,7 +100,7 @@ def test_vec(self):
             c.execute("insert into test_vec(vec1) values (%s)", [np.array([2, 3, 5, 7])])
             c.execute("select * from test_vec")
             r = c.fetchone()
-            self.assertEqual(str(r[1]), str(np.array([2, 3, 5, 7], dtype=np.float32)))
+            self.assertEqual(r[1].all(), np.array([2, 3, 5, 7], dtype=np.float32).all())
         finally:
             c.execute("drop table test_vec")