Skip to content

Commit d408f7a

Browse files
committed
Added test case for an enum inside of an array
1 parent eb0caa7 commit d408f7a

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/test_codecs.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1053,3 +1053,16 @@ async def test_enum(self):
10531053
DROP TABLE tab;
10541054
DROP TYPE enum_t;
10551055
''')
1056+
1057+
async def test_enum_in_array(self):
1058+
await self.con.execute('''
1059+
CREATE TYPE enum_t AS ENUM ('abc', 'def', 'ghi');
1060+
''')
1061+
result = await self.con.fetchrow('''SELECT ARRAY[$1::enum_t];''',
1062+
'abc')
1063+
1064+
self.assertEqual(result, ['abc'])
1065+
1066+
await self.con.execute('''
1067+
DROP TYPE enum_t;
1068+
''')

0 commit comments

Comments
 (0)