diff options
| author | Federico Caselli <cfederico87@gmail.com> | 2021-11-21 21:17:27 +0100 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-11-24 22:51:27 -0500 |
| commit | 31acba8ff7c123a20ae308b7f4ab6df3df264b48 (patch) | |
| tree | a4c39a2123e1b95edf17995ba85bb69ee619f6e4 /test/dialect/postgresql | |
| parent | d3a4e96196cd47858de072ae589c6554088edc24 (diff) | |
| download | sqlalchemy-31acba8ff7c123a20ae308b7f4ab6df3df264b48.tar.gz | |
Clean up most py3k compat
Change-Id: I8172fdcc3103ff92aa049827728484c8779af6b7
Diffstat (limited to 'test/dialect/postgresql')
| -rw-r--r-- | test/dialect/postgresql/test_compiler.py | 10 | ||||
| -rw-r--r-- | test/dialect/postgresql/test_dialect.py | 16 | ||||
| -rw-r--r-- | test/dialect/postgresql/test_types.py | 118 |
3 files changed, 63 insertions, 81 deletions
diff --git a/test/dialect/postgresql/test_compiler.py b/test/dialect/postgresql/test_compiler.py index 7e91f0ebb..a04e2932a 100644 --- a/test/dialect/postgresql/test_compiler.py +++ b/test/dialect/postgresql/test_compiler.py @@ -56,7 +56,6 @@ from sqlalchemy.testing.assertions import AssertsCompiledSQL from sqlalchemy.testing.assertions import expect_warnings from sqlalchemy.testing.assertions import is_ from sqlalchemy.util import OrderedDict -from sqlalchemy.util import u class SequenceTest(fixtures.TestBase, AssertsCompiledSQL): @@ -190,15 +189,10 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL): def test_create_drop_enum(self): # test escaping and unicode within CREATE TYPE for ENUM - typ = postgresql.ENUM( - "val1", "val2", "val's 3", u("méil"), name="myname" - ) + typ = postgresql.ENUM("val1", "val2", "val's 3", "méil", name="myname") self.assert_compile( postgresql.CreateEnumType(typ), - u( - "CREATE TYPE myname AS " - "ENUM ('val1', 'val2', 'val''s 3', 'méil')" - ), + "CREATE TYPE myname AS ENUM ('val1', 'val2', 'val''s 3', 'méil')", ) typ = postgresql.ENUM("val1", "val2", "val's 3", name="PleaseQuoteMe") diff --git a/test/dialect/postgresql/test_dialect.py b/test/dialect/postgresql/test_dialect.py index 7d4ea2ced..57682686c 100644 --- a/test/dialect/postgresql/test_dialect.py +++ b/test/dialect/postgresql/test_dialect.py @@ -51,8 +51,6 @@ from sqlalchemy.testing.assertions import AssertsExecutionResults from sqlalchemy.testing.assertions import eq_ from sqlalchemy.testing.assertions import eq_regex from sqlalchemy.testing.assertions import ne_ -from sqlalchemy.util import u -from sqlalchemy.util import ue if True: from sqlalchemy.dialects.postgresql.psycopg2 import ( @@ -313,10 +311,10 @@ class ExecuteManyMode: ) Table( - u("Unitéble2"), + "Unitéble2", metadata, - Column(u("méil"), Integer, primary_key=True), - Column(ue("\u6e2c\u8a66"), Integer), + Column("méil", Integer, primary_key=True), + Column("\u6e2c\u8a66", Integer), ) def test_insert(self, connection): @@ -478,16 +476,16 @@ class ExecuteManyMode: ) def test_insert_unicode_keys(self, connection): - table = self.tables[u("Unitéble2")] + table = self.tables["Unitéble2"] stmt = table.insert() connection.execute( stmt, [ - {u("méil"): 1, ue("\u6e2c\u8a66"): 1}, - {u("méil"): 2, ue("\u6e2c\u8a66"): 2}, - {u("méil"): 3, ue("\u6e2c\u8a66"): 3}, + {"méil": 1, "\u6e2c\u8a66": 1}, + {"méil": 2, "\u6e2c\u8a66": 2}, + {"méil": 3, "\u6e2c\u8a66": 3}, ], ) diff --git a/test/dialect/postgresql/test_types.py b/test/dialect/postgresql/test_types.py index 4c0b91f93..acca0926e 100644 --- a/test/dialect/postgresql/test_types.py +++ b/test/dialect/postgresql/test_types.py @@ -1,6 +1,7 @@ # coding: utf-8 import datetime import decimal +from enum import Enum as _PY_Enum import re import uuid @@ -319,30 +320,30 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): Column( "value", Enum( - util.u("réveillé"), - util.u("drôle"), - util.u("S’il"), + "réveillé", + "drôle", + "S’il", name="onetwothreetype", ), ), ) metadata.create_all(connection) - connection.execute(t1.insert(), dict(value=util.u("drôle"))) - connection.execute(t1.insert(), dict(value=util.u("réveillé"))) - connection.execute(t1.insert(), dict(value=util.u("S’il"))) + connection.execute(t1.insert(), dict(value="drôle")) + connection.execute(t1.insert(), dict(value="réveillé")) + connection.execute(t1.insert(), dict(value="S’il")) eq_( connection.execute(t1.select().order_by(t1.c.id)).fetchall(), [ - (1, util.u("drôle")), - (2, util.u("réveillé")), - (3, util.u("S’il")), + (1, "drôle"), + (2, "réveillé"), + (3, "S’il"), ], ) m2 = MetaData() t2 = Table("table", m2, autoload_with=connection) eq_( t2.c.value.type.enums, - [util.u("réveillé"), util.u("drôle"), util.u("S’il")], + ["réveillé", "drôle", "S’il"], ) def test_non_native_enum(self, metadata, connection): @@ -390,7 +391,7 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): "bar", Enum( "B", - util.u("Ü"), + "Ü", name="myenum", create_constraint=True, native_enum=False, @@ -406,18 +407,16 @@ class EnumTest(fixtures.TestBase, AssertsExecutionResults): go, [ ( - util.u( - "CREATE TABLE foo (\tbar " - "VARCHAR(1), \tCONSTRAINT myenum CHECK " - "(bar IN ('B', 'Ü')))" - ), + "CREATE TABLE foo (\tbar " + "VARCHAR(1), \tCONSTRAINT myenum CHECK " + "(bar IN ('B', 'Ü')))", {}, ) ], ) - connection.execute(t1.insert(), {"bar": util.u("Ü")}) - eq_(connection.scalar(select(t1.c.bar)), util.u("Ü")) + connection.execute(t1.insert(), {"bar": "Ü"}) + eq_(connection.scalar(select(t1.c.bar)), "Ü") def test_disable_create(self, metadata, connection): metadata = self.metadata @@ -1628,13 +1627,13 @@ class ArrayRoundTripTest: arrtable.insert(), dict( intarr=[1, 2, 3], - strarr=[util.u("abc"), util.u("def")], + strarr=["abc", "def"], ), ) results = connection.execute(arrtable.select()).fetchall() eq_(len(results), 1) eq_(results[0].intarr, [1, 2, 3]) - eq_(results[0].strarr, [util.u("abc"), util.u("def")]) + eq_(results[0].strarr, ["abc", "def"]) def test_insert_array_w_null(self, connection): arrtable = self.tables.arrtable @@ -1642,13 +1641,13 @@ class ArrayRoundTripTest: arrtable.insert(), dict( intarr=[1, None, 3], - strarr=[util.u("abc"), None], + strarr=["abc", None], ), ) results = connection.execute(arrtable.select()).fetchall() eq_(len(results), 1) eq_(results[0].intarr, [1, None, 3]) - eq_(results[0].strarr, [util.u("abc"), None]) + eq_(results[0].strarr, ["abc", None]) def test_array_where(self, connection): arrtable = self.tables.arrtable @@ -1656,11 +1655,11 @@ class ArrayRoundTripTest: arrtable.insert(), dict( intarr=[1, 2, 3], - strarr=[util.u("abc"), util.u("def")], + strarr=["abc", "def"], ), ) connection.execute( - arrtable.insert(), dict(intarr=[4, 5, 6], strarr=util.u("ABC")) + arrtable.insert(), dict(intarr=[4, 5, 6], strarr="ABC") ) results = connection.execute( arrtable.select().where(arrtable.c.intarr == [1, 2, 3]) @@ -1672,7 +1671,7 @@ class ArrayRoundTripTest: arrtable = self.tables.arrtable connection.execute( arrtable.insert(), - dict(intarr=[1, 2, 3], strarr=[util.u("abc"), util.u("def")]), + dict(intarr=[1, 2, 3], strarr=["abc", "def"]), ) results = connection.execute( select(arrtable.c.intarr + [4, 5, 6]) @@ -1684,9 +1683,7 @@ class ArrayRoundTripTest: arrtable = self.tables.arrtable connection.execute( arrtable.insert(), - dict( - id=5, intarr=[1, 2, 3], strarr=[util.u("abc"), util.u("def")] - ), + dict(id=5, intarr=[1, 2, 3], strarr=["abc", "def"]), ) results = connection.execute( select(arrtable.c.id).where(arrtable.c.intarr < [4, 5, 6]) @@ -1700,24 +1697,24 @@ class ArrayRoundTripTest: arrtable.insert(), dict( intarr=[4, 5, 6], - strarr=[[util.ue("m\xe4\xe4")], [util.ue("m\xf6\xf6")]], + strarr=[["m\xe4\xe4"], ["m\xf6\xf6"]], ), ) connection.execute( arrtable.insert(), dict( intarr=[1, 2, 3], - strarr=[util.ue("m\xe4\xe4"), util.ue("m\xf6\xf6")], + strarr=["m\xe4\xe4", "m\xf6\xf6"], ), ) results = connection.execute( arrtable.select().order_by(arrtable.c.intarr) ).fetchall() eq_(len(results), 2) - eq_(results[0].strarr, [util.ue("m\xe4\xe4"), util.ue("m\xf6\xf6")]) + eq_(results[0].strarr, ["m\xe4\xe4", "m\xf6\xf6"]) eq_( results[1].strarr, - [[util.ue("m\xe4\xe4")], [util.ue("m\xf6\xf6")]], + [["m\xe4\xe4"], ["m\xf6\xf6"]], ) def test_array_literal_roundtrip(self, connection): @@ -1790,7 +1787,7 @@ class ArrayRoundTripTest: arrtable.insert(), dict( intarr=[4, 5, 6], - strarr=[util.u("abc"), util.u("def")], + strarr=["abc", "def"], ), ) eq_(connection.scalar(select(arrtable.c.intarr[2:3])), [5, 6]) @@ -1909,11 +1906,11 @@ class ArrayRoundTripTest: def unicode_values(x): return [ - util.u("réveillé"), - util.u("drôle"), - util.u("S’il %s" % x), - util.u("🐍 %s" % x), - util.u("« S’il vous"), + "réveillé", + "drôle", + "S’il %s" % x, + "🐍 %s" % x, + "« S’il vous", ] def json_values(x): @@ -2328,6 +2325,11 @@ class ArrayEnum(fixtures.TestBase): @testing.fixture def array_of_enum_fixture(self, metadata, connection): def go(array_cls, enum_cls): + class MyEnum(_PY_Enum): + a = "aaa" + b = "bbb" + c = "ccc" + tbl = Table( "enum_table", metadata, @@ -2336,23 +2338,11 @@ class ArrayEnum(fixtures.TestBase): "enum_col", array_cls(enum_cls("foo", "bar", "baz", name="an_enum")), ), + Column( + "pyenum_col", + array_cls(enum_cls(MyEnum)), + ), ) - if util.py3k: - from enum import Enum - - class MyEnum(Enum): - a = "aaa" - b = "bbb" - c = "ccc" - - tbl.append_column( - Column( - "pyenum_col", - array_cls(enum_cls(MyEnum)), - ), - ) - else: - MyEnum = None metadata.create_all(connection) connection.execute( @@ -3151,16 +3141,16 @@ class HStoreRoundTripTest(fixtures.TablesTest): def _test_unicode_round_trip(self, connection): s = select( hstore( - array([util.u("réveillé"), util.u("drôle"), util.u("S’il")]), - array([util.u("réveillé"), util.u("drôle"), util.u("S’il")]), + array(["réveillé", "drôle", "S’il"]), + array(["réveillé", "drôle", "S’il"]), ) ) eq_( connection.scalar(s), { - util.u("réveillé"): util.u("réveillé"), - util.u("drôle"): util.u("drôle"), - util.u("S’il"): util.u("S’il"), + "réveillé": "réveillé", + "drôle": "drôle", + "S’il": "S’il", }, ) @@ -3826,7 +3816,7 @@ class JSONRoundTripTest(fixtures.TablesTest): result = connection.execute( select(data_table.c.data["k1"].astext) ).first() - assert isinstance(result[0], util.text_type) + assert isinstance(result[0], str) def test_query_returned_as_int(self, connection): self._fixture_data(connection) @@ -3854,8 +3844,8 @@ class JSONRoundTripTest(fixtures.TablesTest): s = select( cast( { - util.u("réveillé"): util.u("réveillé"), - "data": {"k1": util.u("drôle")}, + "réveillé": "réveillé", + "data": {"k1": "drôle"}, }, self.data_type, ) @@ -3863,8 +3853,8 @@ class JSONRoundTripTest(fixtures.TablesTest): eq_( connection.scalar(s), { - util.u("réveillé"): util.u("réveillé"), - "data": {"k1": util.u("drôle")}, + "réveillé": "réveillé", + "data": {"k1": "drôle"}, }, ) |
