Test fetchone advancing cursor

Author: yunyu

tests/test_dbapi.py

@@ -40,10 +40,24 @@ def test_insert_and_read_data(self):
         # Insert multiple values
         cur.executemany("INSERT INTO rate VALUES (%s, %s)", [("2021-01-02", 72), ("2021-01-03", 96)])
 
-        # Read values
+        # Test fetchone
+        cur.execute("SELECT value FROM rate ORDER BY day DESC LIMIT 2")
+        row1 = cur.fetchone()
+        self.assertEqual(row1, (96,))
+        row2 = cur.fetchone()
+        self.assertEqual(row2, (72,))
+
+        # Test fetchmany
+        cur.execute("SELECT value FROM rate ORDER BY day DESC")
+        result_set1 = cur.fetchmany(2)
+        self.assertEqual(result_set1, ((96,), (72,)))
+        result_set2 = cur.fetchmany(1)
+        self.assertEqual(result_set2, ((24,),))
+
+        # Test fetchall
         cur.execute("SELECT value FROM rate ORDER BY day DESC")
         rows = cur.fetchall()
-        assert rows==((96,), (72,), (24,))
+        self.assertEqual(rows, ((96,), (72,), (24,)))
 
     def test_select_chdb_version(self):
         ver = dbapi.get_client_info()  # chDB version liek '0.12.0'