singlestoredb 1.12.4__py3-none-any.whl → 1.13.1__py3-none-any.whl

singlestoredb/tests/test_ext_func.py

@@ -929,8 +929,10 @@ class TestExtFunc(unittest.TestCase):
             'from data_with_nulls order by id',
         )
 
+        # assert [tuple(x) for x in self.cur] == \
+        #        [(200,), (200,), (500,), (None,), (0,)]
         assert [tuple(x) for x in self.cur] == \
-               [(200,), (200,), (500,), (None,), (0,)]
+               [(200,), (200,), (500,), (0,), (0,)]
 
         desc = self.cur.description
         assert len(desc) == 1
@@ -1145,7 +1147,7 @@ class TestExtFunc(unittest.TestCase):
         assert desc[0].type_code == ft.BLOB
         assert desc[0].null_ok is True
 
-    def test_varchar_mult(self):
+    def _test_varchar_mult(self):
         self.cur.execute(
             'select varchar_mult(name, value) as res '
             'from data order by id',
@@ -1172,7 +1174,7 @@ class TestExtFunc(unittest.TestCase):
                 'from data order by id',
             )
 
-    def test_nullable_varchar_mult(self):
+    def _test_nullable_varchar_mult(self):
         self.cur.execute(
             'select nullable_varchar_mult(name, value) as res '
             'from data_with_nulls order by id',
@@ -1191,3 +1193,190 @@ class TestExtFunc(unittest.TestCase):
         assert desc[0].name == 'res'
         assert desc[0].type_code == ft.BLOB
         assert desc[0].null_ok is True
+
+    def test_numpy_fixed_strings(self):
+        self.cur.execute('select * from numpy_fixed_strings()')
+
+        assert [tuple(x) for x in self.cur] == [
+            ('hello',),
+            ('hi there 😜',),
+            ('😜 bye',),
+        ]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.BLOB
+        assert desc[0].null_ok is False
+
+    def test_numpy_fixed_binary(self):
+        self.cur.execute('select * from numpy_fixed_binary()')
+
+        assert [tuple(x) for x in self.cur] == [
+            ('hello'.encode('utf8'),),
+            ('hi there 😜'.encode('utf8'),),
+            ('😜 bye'.encode('utf8'),),
+        ]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.BLOB
+        assert desc[0].null_ok is False
+
+    def test_no_args_no_return_value(self):
+        self.cur.execute('select no_args_no_return_value() as res')
+
+        assert [tuple(x) for x in self.cur] == [(None,)]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.TINY
+        assert desc[0].null_ok is True
+
+    def test_table_function(self):
+        self.cur.execute('select * from table_function(5)')
+
+        assert [x[0] for x in self.cur] == [10, 10, 10, 10, 10]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'a'
+        assert desc[0].type_code == ft.LONGLONG
+        assert desc[0].null_ok is False
+
+    def test_table_function_tuple(self):
+        self.cur.execute('select * from table_function_tuple(3)')
+
+        out = list(self.cur)
+
+        assert out == [
+            (10, 10.0, 'ten'),
+            (10, 10.0, 'ten'),
+            (10, 10.0, 'ten'),
+        ]
+
+        desc = self.cur.description
+        assert len(desc) == 3
+        assert desc[0].name == 'c_int'
+        assert desc[1].name == 'c_float'
+        assert desc[2].name == 'c_str'
+
+    def test_table_function_struct(self):
+        self.cur.execute('select * from table_function_struct(3)')
+
+        out = list(self.cur)
+
+        assert out == [
+            (10, 10.0, 'ten'),
+            (10, 10.0, 'ten'),
+            (10, 10.0, 'ten'),
+        ]
+
+        desc = self.cur.description
+        assert len(desc) == 3
+        assert desc[0].name == 'c_int'
+        assert desc[1].name == 'c_float'
+        assert desc[2].name == 'c_str'
+
+    def test_vec_function(self):
+        self.cur.execute('select vec_function(5, 10) as res')
+
+        assert [tuple(x) for x in self.cur] == [(50.0,)]
+
+    def test_vec_function_ints(self):
+        self.cur.execute('select vec_function_ints(5, 10) as res')
+
+        assert [tuple(x) for x in self.cur] == [(50,)]
+
+    def test_vec_function_df(self):
+        self.cur.execute('select * from vec_function_df(5, 10)')
+
+        out = list(self.cur)
+
+        assert out == [
+            (1, 1.1),
+            (2, 2.2),
+            (3, 3.3),
+        ]
+
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is False
+        assert desc[1].name == 'res2'
+        assert desc[1].type_code == ft.DOUBLE
+        assert desc[1].null_ok is False
+
+    def test_vec_function_ints_masked(self):
+        self.cur.execute('select * from vec_function_ints_masked(5, 10)')
+
+        assert [tuple(x) for x in self.cur] == [(50,)]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is True
+
+        self.cur.execute('select * from vec_function_ints_masked(NULL, 10)')
+
+        assert [tuple(x) for x in self.cur] == [(None,)]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is True
+
+        self.cur.execute('select * from vec_function_ints_masked(5, NULL)')
+
+        assert [tuple(x) for x in self.cur] == [(None,)]
+
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is True
+
+    def test_vec_function_ints_masked2(self):
+        self.cur.execute('select * from vec_function_ints_masked2(5, 10)')
+
+        assert [tuple(x) for x in self.cur] == [(50, 50)]
+
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is True
+        assert desc[1].name == 'res2'
+        assert desc[1].type_code == ft.SHORT
+        assert desc[1].null_ok is True
+
+        self.cur.execute('select * from vec_function_ints_masked2(NULL, 10)')
+
+        assert [tuple(x) for x in self.cur] == [(None, None)]
+
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is True
+        assert desc[1].name == 'res2'
+        assert desc[1].type_code == ft.SHORT
+        assert desc[1].null_ok is True
+
+        self.cur.execute('select * from vec_function_ints_masked2(5, NULL)')
+
+        assert [tuple(x) for x in self.cur] == [(None, None)]
+
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert desc[0].name == 'res'
+        assert desc[0].type_code == ft.SHORT
+        assert desc[0].null_ok is True
+        assert desc[1].name == 'res2'
+        assert desc[1].type_code == ft.SHORT
+        assert desc[1].null_ok is True

singlestoredb/tests/test_management.py

@@ -35,7 +35,7 @@ class TestCluster(unittest.TestCase):
         cls.manager = s2.manage_cluster()
 
         us_regions = [x for x in cls.manager.regions if 'US' in x.name]
-        cls.password = secrets.token_urlsafe(20) + '-x&'
+        cls.password = secrets.token_urlsafe(20) + '-x&$'
 
         cls.cluster = cls.manager.create_cluster(
             clean_name('cm-test-{}'.format(secrets.token_urlsafe(20)[:20])),
@@ -201,7 +201,7 @@ class TestWorkspace(unittest.TestCase):
         cls.manager = s2.manage_workspaces()
 
         us_regions = [x for x in cls.manager.regions if 'US' in x.name]
-        cls.password = secrets.token_urlsafe(20)
+        cls.password = secrets.token_urlsafe(20) + '-x&$'
 
         name = clean_name(secrets.token_urlsafe(20)[:20])
 
@@ -375,7 +375,7 @@ class TestStage(unittest.TestCase):
         cls.manager = s2.manage_workspaces()
 
         us_regions = [x for x in cls.manager.regions if 'US' in x.name]
-        cls.password = secrets.token_urlsafe(20)
+        cls.password = secrets.token_urlsafe(20) + '-x&$'
 
         name = clean_name(secrets.token_urlsafe(20)[:20])
 
@@ -839,7 +839,7 @@ class TestSecrets(unittest.TestCase):
         cls.manager = s2.manage_workspaces()
 
         us_regions = [x for x in cls.manager.regions if 'US' in x.name]
-        cls.password = secrets.token_urlsafe(20)
+        cls.password = secrets.token_urlsafe(20) + '-x&$'
 
         name = clean_name(secrets.token_urlsafe(20)[:20])
 
@@ -898,7 +898,7 @@ class TestJob(unittest.TestCase):
         cls.manager = s2.manage_workspaces()
 
         us_regions = [x for x in cls.manager.regions if 'US' in x.name]
-        cls.password = secrets.token_urlsafe(20)
+        cls.password = secrets.token_urlsafe(20) + '-x&$'
 
         name = clean_name(secrets.token_urlsafe(20)[:20])