database/postgres
database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-04-18 13:44:19 +03:00
Code

plpython: Add test for returning Python set from SETOF function

Browse Source 
This is claimed in the documentation but there was a no test case for
it.

Reported-by: Bogdan Grigorenko <gri.bogdan.2020@gmail.com>
Discussion: https://www.postgresql.org/message-id/flat/173543330569.680.6706329879058172623%40wrigleys.postgresql.org
This commit is contained in:
Peter Eisentraut 2025-04-03 11:04:28 +02:00
parent d1d83827ba
commit 231064aa0f
2 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/pl/plpython/expected/plpython_setof.out
View File

@ -17,6 +17,12 @@ for i in range(count):
t += ( content, ) t += ( content, )
return t return t
$$ LANGUAGE plpython3u; $$ LANGUAGE plpython3u;
CREATE FUNCTION test_setof_as_set(count integer, content text) RETURNS SETOF text AS $$
s = set()
for i in range(count):
s.add(content * (i + 1) if content is not None else None)
return s
$$ LANGUAGE plpython3u;
CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$ CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$
class producer: class producer:
def __init__ (self, icount, icontent): def __init__ (self, icount, icontent):
@ -90,6 +96,30 @@ SELECT test_setof_as_tuple(2, null);
(2 rows) (2 rows)
SELECT * FROM test_setof_as_set(0, 'set') ORDER BY 1;
test_setof_as_set
-------------------
(0 rows)
SELECT * FROM test_setof_as_set(1, 'set') ORDER BY 1;
test_setof_as_set
-------------------
set
(1 row)
SELECT * FROM test_setof_as_set(2, 'set') ORDER BY 1;
test_setof_as_set
-------------------
set
setset
(2 rows)
SELECT * FROM test_setof_as_set(2, null) ORDER BY 1;
test_setof_as_set
-------------------
(1 row)
SELECT test_setof_as_iterator(0, 'list'); SELECT test_setof_as_iterator(0, 'list');
test_setof_as_iterator test_setof_as_iterator
------------------------ ------------------------

12
src/pl/plpython/sql/plpython_setof.sql
View File

@ -20,6 +20,13 @@ for i in range(count):
return t return t
$$ LANGUAGE plpython3u; $$ LANGUAGE plpython3u;
CREATE FUNCTION test_setof_as_set(count integer, content text) RETURNS SETOF text AS $$
s = set()
for i in range(count):
s.add(content * (i + 1) if content is not None else None)
return s
$$ LANGUAGE plpython3u;
CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$ CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$
class producer: class producer:
def __init__ (self, icount, icontent): def __init__ (self, icount, icontent):
@ -56,6 +63,11 @@ SELECT test_setof_as_tuple(1, 'tuple');
SELECT test_setof_as_tuple(2, 'tuple'); SELECT test_setof_as_tuple(2, 'tuple');
SELECT test_setof_as_tuple(2, null); SELECT test_setof_as_tuple(2, null);
SELECT * FROM test_setof_as_set(0, 'set') ORDER BY 1;
SELECT * FROM test_setof_as_set(1, 'set') ORDER BY 1;
SELECT * FROM test_setof_as_set(2, 'set') ORDER BY 1;
SELECT * FROM test_setof_as_set(2, null) ORDER BY 1;
SELECT test_setof_as_iterator(0, 'list'); SELECT test_setof_as_iterator(0, 'list');
SELECT test_setof_as_iterator(1, 'list'); SELECT test_setof_as_iterator(1, 'list');
SELECT test_setof_as_iterator(2, 'list'); SELECT test_setof_as_iterator(2, 'list');