]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
test(enum): drop unneeded cursor creation
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 21 Apr 2022 01:48:00 +0000 (03:48 +0200)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Fri, 22 Apr 2022 03:03:24 +0000 (05:03 +0200)
tests/types/test_enum.py

index dac41c4fe63e2587dcba32a3d1be0c63a5badca1..90cc00d9f4de4335a8274df0b84e16df110d4589 100644 (file)
@@ -229,10 +229,9 @@ async def test_enum_async(aconn, testenum, encoding, fmt_in, fmt_out):
     name, enum, labels = testenum
     register_enum(await EnumInfo.fetch(aconn, name), enum, aconn)
 
-    async with aconn.cursor(binary=fmt_out) as cur:
-        for label in labels:
-            cur = await cur.execute(f"select %{fmt_in}::{name}", [label])
-            assert (await cur.fetchone())[0] == enum[label]
+    for label in labels:
+        cur = await aconn.execute(f"select %{fmt_in}::{name}", [label], binary=fmt_out)
+        assert (await cur.fetchone())[0] == enum[label]
 
-        cur = await cur.execute(f"select %{fmt_in}", [list(enum)])
-        assert (await cur.fetchone())[0] == list(enum)
+    cur = await cur.execute(f"select %{fmt_in}", [list(enum)])
+    assert (await cur.fetchone())[0] == list(enum)