with psycopg.ServerCursor(conn, "baz", row_factory=rows.namedtuple_row) as cur:
assert cur.name == "baz"
- assert cur.row_factory is rows.namedtuple_row # type: ignore
+ assert cur.row_factory is rows.namedtuple_row
def test_init_params(conn):
aconn, "baz", row_factory=rows.namedtuple_row
) as cur:
assert cur.name == "baz"
- assert cur.row_factory is rows.namedtuple_row # type: ignore
+ assert cur.row_factory is rows.namedtuple_row
async def test_init_params(aconn):
def test_remap_rename(conn):
- enum = Enum("RenamedEnum", "FOO BAR QUX")
+ RenamedEnum = Enum("RenamedEnum", "FOO BAR QUX")
+ enum = RenamedEnum
info = EnumInfo.fetch(conn, "puretestenum")
register_enum(info, conn, enum, mapping={enum.QUX: "BAZ"})
def test_remap_more_python(conn):
- enum = Enum("LargerEnum", "FOO BAR BAZ QUX QUUX QUUUX")
+ LargerEnum = Enum("LargerEnum", "FOO BAR BAZ QUX QUUX QUUUX")
+ enum = LargerEnum
info = EnumInfo.fetch(conn, "puretestenum")
mapping = {enum[m]: "BAZ" for m in ["QUX", "QUUX", "QUUUX"]}
register_enum(info, conn, enum, mapping=mapping)
def test_remap_more_postgres(conn):
- enum = Enum("SmallerEnum", "FOO")
+ SmallerEnum = Enum("SmallerEnum", "FOO")
+ enum = SmallerEnum
info = EnumInfo.fetch(conn, "puretestenum")
mapping = [(enum.FOO, "BAR"), (enum.FOO, "BAZ")]
register_enum(info, conn, enum, mapping=mapping)