singlestoredb 0.4.0__py3-none-any.whl → 1.0.4__py3-none-any.whl

singlestoredb/{clients/pymysqlsv → mysql}/tests/test_basic.py

@@ -5,22 +5,25 @@ import time
 
 import pytest
 
-import singlestoredb.clients.pymysqlsv as sv
-import singlestoredb.clients.pymysqlsv.cursors as cursors
-from singlestoredb.clients.pymysqlsv.err import ProgrammingError
-from singlestoredb.clients.pymysqlsv.tests import base
+import singlestoredb.mysql as sv
+import singlestoredb.mysql.cursors as cursors
+from singlestoredb.mysql.tests import base
 
 
 __all__ = ['TestConversion', 'TestCursor', 'TestBulkInserts']
 
 
 class TestConversion(base.PyMySQLTestCase):
+
     def test_datatypes(self):
-        """test every data type"""
+        """Test every data type."""
         conn = self.connect()
         c = conn.cursor()
         c.execute(
-            'create table test_datatypes (b bit, i int, l bigint, f real, s varchar(32), u varchar(32), bb blob, d date, dt datetime, ts timestamp, td time, t time, st datetime)',
+            'create table test_datatypes (b bit, i int, l bigint, f real, '
+            '                             s varchar(32), u varchar(32), bb blob, '
+            '                             d date, dt datetime, ts timestamp, '
+            '                             td time, t time, st datetime)',
         )
         try:
             # insert values
@@ -40,7 +43,8 @@ class TestConversion(base.PyMySQLTestCase):
                 time.localtime(),
             )
             c.execute(
-                'insert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)',
+                'insert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) '
+                '                    values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)',
                 v,
             )
             c.execute('select b,i,l,f,s,u,bb,d,dt,td,t,st from test_datatypes')
@@ -56,7 +60,8 @@ class TestConversion(base.PyMySQLTestCase):
 
             # check nulls
             c.execute(
-                'insert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)',
+                'insert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) '
+                '                    values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)',
                 [None] * 12,
             )
             c.execute('select b,i,l,f,s,u,bb,d,dt,td,t,st from test_datatypes')
@@ -75,14 +80,15 @@ class TestConversion(base.PyMySQLTestCase):
                     'select l from test_datatypes where i in %s order by i', (seq,),
                 )
                 r = c.fetchall()
-                self.assertEqual(((4,), (8,)), r)
+                # NOTE: C extension returns a list, not a tuple
+                self.assertEqual(((4,), (8,)), tuple(r))
                 c.execute('delete from test_datatypes')
 
         finally:
             c.execute('drop table test_datatypes')
 
     def test_dict(self):
-        """test dict escaping"""
+        """Test dict escaping."""
         conn = self.connect()
         c = conn.cursor()
         c.execute('create table test_dict (a integer, b integer, c integer)')
@@ -121,7 +127,7 @@ class TestConversion(base.PyMySQLTestCase):
             c.execute('drop table test_dict')
 
     def test_binary(self):
-        """test binary data"""
+        """Test binary data."""
         data = bytes(bytearray(range(255)))
         conn = self.connect()
         self.safe_create_table(
@@ -134,7 +140,7 @@ class TestConversion(base.PyMySQLTestCase):
             self.assertEqual(data, c.fetchone()[0])
 
     def test_blob(self):
-        """test blob data"""
+        """Test blob data."""
         data = bytes(bytearray(range(256)) * 4)
         conn = self.connect()
         self.safe_create_table(conn, 'test_blob', 'create table test_blob (b blob)')
@@ -145,7 +151,7 @@ class TestConversion(base.PyMySQLTestCase):
             self.assertEqual(data, c.fetchone()[0])
 
     def test_untyped(self):
-        """test conversion of null, empty string"""
+        """Test conversion of null, empty string."""
         conn = self.connect()
         c = conn.cursor()
         c.execute("select null,''")
@@ -154,11 +160,13 @@ class TestConversion(base.PyMySQLTestCase):
         self.assertEqual(('', None), c.fetchone())
 
     def test_timedelta(self):
-        """test timedelta conversion"""
+        """Test timedelta conversion."""
         conn = self.connect()
         c = conn.cursor()
         c.execute(
-            "select time('12:30'), time('23:12:59'), time('23:12:59.05100'), time('-12:30'), time('-23:12:59'), time('-23:12:59.05100'), time('-00:30')",
+            "select time('12:30'), time('23:12:59'), time('23:12:59.05100'), "
+            "       time('-12:30'), time('-23:12:59'), time('-23:12:59.05100'), "
+            "       time('-00:30')",
         )
         self.assertEqual(
             (
@@ -174,7 +182,7 @@ class TestConversion(base.PyMySQLTestCase):
         )
 
     def test_datetime_microseconds(self):
-        """test datetime conversion w microseconds"""
+        """Test datetime conversion w microseconds."""
 
         conn = self.connect()
         c = conn.cursor()
@@ -243,7 +251,7 @@ class TestCursor(base.PyMySQLTestCase):
     #    self.assertEqual(r, c.description)
 
     def test_fetch_no_result(self):
-        """test a fetchone() with no rows"""
+        """Test a fetchone() with no rows."""
         conn = self.connect()
         c = conn.cursor()
         c.execute('create table test_nr (b varchar(32))')
@@ -255,7 +263,7 @@ class TestCursor(base.PyMySQLTestCase):
             c.execute('drop table test_nr')
 
     def test_aggregates(self):
-        """test aggregate functions"""
+        """Test aggregate functions."""
         conn = self.connect()
         c = conn.cursor()
         try:
@@ -269,7 +277,7 @@ class TestCursor(base.PyMySQLTestCase):
             c.execute('drop table test_aggregates')
 
     def test_single_tuple(self):
-        """test a single tuple"""
+        """Test a single tuple."""
         conn = self.connect()
         c = conn.cursor()
         self.safe_create_table(
@@ -306,12 +314,12 @@ create table test_json (
         cur.execute('INSERT INTO test_json (id, `json`) values (42, %s)', (json_str,))
         cur.execute('SELECT `json` from `test_json` WHERE `id`=42')
         res = cur.fetchone()[0]
-        self.assertEqual(json.loads(res), json.loads(json_str))
+        self.assertEqual(res, json.loads(json_str))
 
 #       cur.execute("SELECT CAST(%s AS JSON) AS x", (json_str,))
         cur.execute('SELECT %s :> JSON AS x', (json_str,))
         res = cur.fetchone()[0]
-        self.assertEqual(json.loads(res), json.loads(json_str))
+        self.assertEqual(res, json.loads(json_str))
 
 
 class TestBulkInserts(base.PyMySQLTestCase):
@@ -320,8 +328,8 @@ class TestBulkInserts(base.PyMySQLTestCase):
 
     def setUp(self):
         super(TestBulkInserts, self).setUp()
-        self.conn = conn = self.connect()
-        c = conn.cursor(self.cursor_type)
+        self.conn = conn = self.connect(cursorclass=self.cursor_type)
+        c = conn.cursor()  # noqa: F841
 
         # create a table ane some data to query
         self.safe_create_table(
@@ -409,7 +417,7 @@ values (0,
         self._verify_records(data)
 
     def test_issue_288(self):
-        """executemany should work with "insert ... on update" """
+        """Executemany should work with "insert ... on update"."""
         conn = self.connect()
         cursor = conn.cursor()
         data = [(0, 'bob', 21, 123), (1, 'jim', 56, 45), (2, 'fred', 100, 180)]