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 /lib/sqlalchemy/testing/suite | |
| parent | d3a4e96196cd47858de072ae589c6554088edc24 (diff) | |
| download | sqlalchemy-31acba8ff7c123a20ae308b7f4ab6df3df264b48.tar.gz | |
Clean up most py3k compat
Change-Id: I8172fdcc3103ff92aa049827728484c8779af6b7
Diffstat (limited to 'lib/sqlalchemy/testing/suite')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_results.py | 3 | ||||
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_select.py | 5 | ||||
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_types.py | 45 | ||||
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_unicode_ddl.py | 120 |
4 files changed, 75 insertions, 98 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_results.py b/lib/sqlalchemy/testing/suite/test_results.py index f470432d2..5ad68034b 100644 --- a/lib/sqlalchemy/testing/suite/test_results.py +++ b/lib/sqlalchemy/testing/suite/test_results.py @@ -14,7 +14,6 @@ from ... import sql from ... import String from ... import testing from ... import text -from ... import util class RowFetchTest(fixtures.TablesTest): @@ -306,7 +305,7 @@ class ServerSideCursorsTest( ): engine = self._fixture(engine_ss_arg) with engine.begin() as conn: - if isinstance(statement, util.string_types): + if isinstance(statement, str): result = conn.exec_driver_sql(statement) else: result = conn.execute(statement) diff --git a/lib/sqlalchemy/testing/suite/test_select.py b/lib/sqlalchemy/testing/suite/test_select.py index b5a3dca3a..c1228f5df 100644 --- a/lib/sqlalchemy/testing/suite/test_select.py +++ b/lib/sqlalchemy/testing/suite/test_select.py @@ -1,3 +1,4 @@ +import collections.abc as collections_abc import itertools from .. import AssertsCompiledSQL @@ -32,11 +33,9 @@ from ... import true from ... import tuple_ from ... import TupleType from ... import union -from ... import util from ... import values from ...exc import DatabaseError from ...exc import ProgrammingError -from ...util import collections_abc class CollateTest(fixtures.TablesTest): @@ -131,7 +130,7 @@ class OrderByLabelTest(fixtures.TablesTest): ly = (func.lower(table.c.q) + table.c.p).label("ly") self._assert_result( select(lx, ly).order_by(lx, ly.desc()), - [(3, util.u("q1p3")), (5, util.u("q2p2")), (7, util.u("q3p1"))], + [(3, "q1p3"), (5, "q2p2"), (7, "q3p1")], ) def test_plain_desc(self): diff --git a/lib/sqlalchemy/testing/suite/test_types.py b/lib/sqlalchemy/testing/suite/test_types.py index 4a5396ed8..7989b1e39 100644 --- a/lib/sqlalchemy/testing/suite/test_types.py +++ b/lib/sqlalchemy/testing/suite/test_types.py @@ -39,10 +39,8 @@ from ... import type_coerce from ... import TypeDecorator from ... import Unicode from ... import UnicodeText -from ... import util from ...orm import declarative_base from ...orm import Session -from ...util import u class _LiteralRoundTripFixture: @@ -93,7 +91,7 @@ class _LiteralRoundTripFixture: class _UnicodeFixture(_LiteralRoundTripFixture, fixtures.TestBase): __requires__ = ("unicode_data",) - data = u( + data = ( "Alors vous imaginez ma đ surprise, au lever du jour, " "quand une drĂŽle de petite đ voix mâa rĂ©veillĂ©. Elle " "disait: « Sâil vous plaĂźt⊠dessine-moi đ un mouton! »" @@ -124,7 +122,7 @@ class _UnicodeFixture(_LiteralRoundTripFixture, fixtures.TestBase): row = connection.execute(select(unicode_table.c.unicode_data)).first() eq_(row, (self.data,)) - assert isinstance(row[0], util.text_type) + assert isinstance(row[0], str) def test_round_trip_executemany(self, connection): unicode_table = self.tables.unicode_table @@ -139,7 +137,7 @@ class _UnicodeFixture(_LiteralRoundTripFixture, fixtures.TestBase): ).fetchall() eq_(rows, [(self.data,) for i in range(1, 4)]) for row in rows: - assert isinstance(row[0], util.text_type) + assert isinstance(row[0], str) def _test_null_strings(self, connection): unicode_table = self.tables.unicode_table @@ -154,18 +152,16 @@ class _UnicodeFixture(_LiteralRoundTripFixture, fixtures.TestBase): unicode_table = self.tables.unicode_table connection.execute( - unicode_table.insert(), {"id": 1, "unicode_data": u("")} + unicode_table.insert(), {"id": 1, "unicode_data": ""} ) row = connection.execute(select(unicode_table.c.unicode_data)).first() - eq_(row, (u(""),)) + eq_(row, ("",)) def test_literal(self, literal_round_trip): literal_round_trip(self.datatype, [self.data], [self.data]) def test_literal_non_ascii(self, literal_round_trip): - literal_round_trip( - self.datatype, [util.u("rĂ©veđ illĂ©")], [util.u("rĂ©veđ illĂ©")] - ) + literal_round_trip(self.datatype, ["rĂ©veđ illĂ©"], ["rĂ©veđ illĂ©"]) class UnicodeVarcharTest(_UnicodeFixture, fixtures.TablesTest): @@ -243,9 +239,7 @@ class TextTest(_LiteralRoundTripFixture, fixtures.TablesTest): literal_round_trip(Text, ["some text"], ["some text"]) def test_literal_non_ascii(self, literal_round_trip): - literal_round_trip( - Text, [util.u("rĂ©veđ illĂ©")], [util.u("rĂ©veđ illĂ©")] - ) + literal_round_trip(Text, ["rĂ©veđ illĂ©"], ["rĂ©veđ illĂ©"]) def test_literal_quoting(self, literal_round_trip): data = """some 'text' hey "hi there" that's text""" @@ -277,9 +271,7 @@ class StringTest(_LiteralRoundTripFixture, fixtures.TestBase): literal_round_trip(String(40), ["some text"], ["some text"]) def test_literal_non_ascii(self, literal_round_trip): - literal_round_trip( - String(40), [util.u("rĂ©veđ illĂ©")], [util.u("rĂ©veđ illĂ©")] - ) + literal_round_trip(String(40), ["rĂ©veđ illĂ©"], ["rĂ©veđ illĂ©"]) def test_literal_quoting(self, literal_round_trip): data = """some 'text' hey "hi there" that's text""" @@ -474,10 +466,7 @@ class IntegerTest(_LiteralRoundTripFixture, fixtures.TestBase): eq_(row, (data,)) - if util.py3k: - assert isinstance(row[0], int) - else: - assert isinstance(row[0], (long, int)) # noqa + assert isinstance(row[0], int) return run @@ -880,10 +869,10 @@ class JSONTest(_LiteralRoundTripFixture, fixtures.TablesTest): ("boolean", None), ("string", "some string"), ("string", None), - ("string", util.u("rĂ©ve illĂ©")), + ("string", "rĂ©ve illĂ©"), ( "string", - util.u("rĂ©veđ illĂ©"), + "rĂ©veđ illĂ©", testing.requires.json_index_supplementary_unicode_element, ), ("integer", 15), @@ -1080,8 +1069,8 @@ class JSONTest(_LiteralRoundTripFixture, fixtures.TablesTest): (-1.0,), (15.052,), ("a string",), - (util.u("rĂ©ve illĂ©"),), - (util.u("rĂ©veđ illĂ©"),), + ("rĂ©ve illĂ©",), + ("rĂ©veđ illĂ©",), ) def test_single_element_round_trip(self, element): data_table = self.tables.data_table @@ -1243,8 +1232,8 @@ class JSONTest(_LiteralRoundTripFixture, fixtures.TablesTest): { "name": "r1", "data": { - util.u("rĂ©veđ illĂ©"): util.u("rĂ©veđ illĂ©"), - "data": {"k1": util.u("drĂŽlđe")}, + "rĂ©veđ illĂ©": "rĂ©veđ illĂ©", + "data": {"k1": "drĂŽlđe"}, }, }, ) @@ -1252,8 +1241,8 @@ class JSONTest(_LiteralRoundTripFixture, fixtures.TablesTest): eq_( conn.scalar(select(self.tables.data_table.c.data)), { - util.u("rĂ©veđ illĂ©"): util.u("rĂ©veđ illĂ©"), - "data": {"k1": util.u("drĂŽlđe")}, + "rĂ©veđ illĂ©": "rĂ©veđ illĂ©", + "data": {"k1": "drĂŽlđe"}, }, ) diff --git a/lib/sqlalchemy/testing/suite/test_unicode_ddl.py b/lib/sqlalchemy/testing/suite/test_unicode_ddl.py index 1334eb8db..a0cd6734a 100644 --- a/lib/sqlalchemy/testing/suite/test_unicode_ddl.py +++ b/lib/sqlalchemy/testing/suite/test_unicode_ddl.py @@ -10,8 +10,6 @@ from sqlalchemy.testing import eq_ from sqlalchemy.testing import fixtures from sqlalchemy.testing.schema import Column from sqlalchemy.testing.schema import Table -from sqlalchemy.util import u -from sqlalchemy.util import ue class UnicodeSchemaTest(fixtures.TablesTest): @@ -23,20 +21,20 @@ class UnicodeSchemaTest(fixtures.TablesTest): global t1, t2, t3 t1 = Table( - u("unitable1"), + "unitable1", 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), test_needs_fk=True, ) t2 = Table( - u("UnitĂ©ble2"), + "UnitĂ©ble2", metadata, - Column(u("mĂ©il"), Integer, primary_key=True, key="a"), + Column("mĂ©il", Integer, primary_key=True, key="a"), Column( - ue("\u6e2c\u8a66"), + "\u6e2c\u8a66", Integer, - ForeignKey(u("unitable1.mĂ©il")), + ForeignKey("unitable1.mĂ©il"), key="b", ), test_needs_fk=True, @@ -45,55 +43,53 @@ class UnicodeSchemaTest(fixtures.TablesTest): # Few DBs support Unicode foreign keys if testing.against("sqlite"): t3 = Table( - ue("\u6e2c\u8a66"), + "\u6e2c\u8a66", metadata, Column( - ue("\u6e2c\u8a66_id"), + "\u6e2c\u8a66_id", Integer, primary_key=True, autoincrement=False, ), Column( - ue("unitable1_\u6e2c\u8a66"), + "unitable1_\u6e2c\u8a66", Integer, - ForeignKey(ue("unitable1.\u6e2c\u8a66")), + ForeignKey("unitable1.\u6e2c\u8a66"), ), + Column("UnitĂ©ble2_b", Integer, ForeignKey("UnitĂ©ble2.b")), Column( - u("UnitĂ©ble2_b"), Integer, ForeignKey(u("UnitĂ©ble2.b")) - ), - Column( - ue("\u6e2c\u8a66_self"), + "\u6e2c\u8a66_self", Integer, - ForeignKey(ue("\u6e2c\u8a66.\u6e2c\u8a66_id")), + ForeignKey("\u6e2c\u8a66.\u6e2c\u8a66_id"), ), test_needs_fk=True, ) else: t3 = Table( - ue("\u6e2c\u8a66"), + "\u6e2c\u8a66", metadata, Column( - ue("\u6e2c\u8a66_id"), + "\u6e2c\u8a66_id", Integer, primary_key=True, autoincrement=False, ), - Column(ue("unitable1_\u6e2c\u8a66"), Integer), - Column(u("UnitĂ©ble2_b"), Integer), - Column(ue("\u6e2c\u8a66_self"), Integer), + Column("unitable1_\u6e2c\u8a66", Integer), + Column("UnitĂ©ble2_b", Integer), + Column("\u6e2c\u8a66_self", Integer), test_needs_fk=True, ) def test_insert(self, connection): - connection.execute(t1.insert(), {u("mĂ©il"): 1, ue("\u6e2c\u8a66"): 5}) - connection.execute(t2.insert(), {u("a"): 1, u("b"): 1}) + connection.execute(t1.insert(), {"mĂ©il": 1, "\u6e2c\u8a66": 5}) + connection.execute(t2.insert(), {"a": 1, "b": 1}) connection.execute( t3.insert(), { - ue("\u6e2c\u8a66_id"): 1, - ue("unitable1_\u6e2c\u8a66"): 5, - u("UnitĂ©ble2_b"): 1, - ue("\u6e2c\u8a66_self"): 1, + "\u6e2c\u8a66_id": 1, + "unitable1_\u6e2c\u8a66": 5, + "UnitĂ©ble2_b": 1, + "\u6e2c\u8a66_self": 1, }, ) @@ -102,42 +98,42 @@ class UnicodeSchemaTest(fixtures.TablesTest): eq_(connection.execute(t3.select()).fetchall(), [(1, 5, 1, 1)]) def test_col_targeting(self, connection): - connection.execute(t1.insert(), {u("mĂ©il"): 1, ue("\u6e2c\u8a66"): 5}) - connection.execute(t2.insert(), {u("a"): 1, u("b"): 1}) + connection.execute(t1.insert(), {"mĂ©il": 1, "\u6e2c\u8a66": 5}) + connection.execute(t2.insert(), {"a": 1, "b": 1}) connection.execute( t3.insert(), { - ue("\u6e2c\u8a66_id"): 1, - ue("unitable1_\u6e2c\u8a66"): 5, - u("UnitĂ©ble2_b"): 1, - ue("\u6e2c\u8a66_self"): 1, + "\u6e2c\u8a66_id": 1, + "unitable1_\u6e2c\u8a66": 5, + "UnitĂ©ble2_b": 1, + "\u6e2c\u8a66_self": 1, }, ) row = connection.execute(t1.select()).first() - eq_(row._mapping[t1.c[u("mĂ©il")]], 1) - eq_(row._mapping[t1.c[ue("\u6e2c\u8a66")]], 5) + eq_(row._mapping[t1.c["mĂ©il"]], 1) + eq_(row._mapping[t1.c["\u6e2c\u8a66"]], 5) row = connection.execute(t2.select()).first() - eq_(row._mapping[t2.c[u("a")]], 1) - eq_(row._mapping[t2.c[u("b")]], 1) + eq_(row._mapping[t2.c["a"]], 1) + eq_(row._mapping[t2.c["b"]], 1) row = connection.execute(t3.select()).first() - eq_(row._mapping[t3.c[ue("\u6e2c\u8a66_id")]], 1) - eq_(row._mapping[t3.c[ue("unitable1_\u6e2c\u8a66")]], 5) - eq_(row._mapping[t3.c[u("UnitĂ©ble2_b")]], 1) - eq_(row._mapping[t3.c[ue("\u6e2c\u8a66_self")]], 1) + eq_(row._mapping[t3.c["\u6e2c\u8a66_id"]], 1) + eq_(row._mapping[t3.c["unitable1_\u6e2c\u8a66"]], 5) + eq_(row._mapping[t3.c["UnitĂ©ble2_b"]], 1) + eq_(row._mapping[t3.c["\u6e2c\u8a66_self"]], 1) def test_reflect(self, connection): - connection.execute(t1.insert(), {u("mĂ©il"): 2, ue("\u6e2c\u8a66"): 7}) - connection.execute(t2.insert(), {u("a"): 2, u("b"): 2}) + connection.execute(t1.insert(), {"mĂ©il": 2, "\u6e2c\u8a66": 7}) + connection.execute(t2.insert(), {"a": 2, "b": 2}) connection.execute( t3.insert(), { - ue("\u6e2c\u8a66_id"): 2, - ue("unitable1_\u6e2c\u8a66"): 7, - u("UnitĂ©ble2_b"): 2, - ue("\u6e2c\u8a66_self"): 2, + "\u6e2c\u8a66_id": 2, + "unitable1_\u6e2c\u8a66": 7, + "UnitĂ©ble2_b": 2, + "\u6e2c\u8a66_self": 2, }, ) @@ -146,42 +142,36 @@ class UnicodeSchemaTest(fixtures.TablesTest): tt2 = Table(t2.name, meta, autoload_with=connection) tt3 = Table(t3.name, meta, autoload_with=connection) - connection.execute(tt1.insert(), {u("mĂ©il"): 1, ue("\u6e2c\u8a66"): 5}) - connection.execute(tt2.insert(), {u("mĂ©il"): 1, ue("\u6e2c\u8a66"): 1}) + connection.execute(tt1.insert(), {"mĂ©il": 1, "\u6e2c\u8a66": 5}) + connection.execute(tt2.insert(), {"mĂ©il": 1, "\u6e2c\u8a66": 1}) connection.execute( tt3.insert(), { - ue("\u6e2c\u8a66_id"): 1, - ue("unitable1_\u6e2c\u8a66"): 5, - u("UnitĂ©ble2_b"): 1, - ue("\u6e2c\u8a66_self"): 1, + "\u6e2c\u8a66_id": 1, + "unitable1_\u6e2c\u8a66": 5, + "UnitĂ©ble2_b": 1, + "\u6e2c\u8a66_self": 1, }, ) eq_( - connection.execute( - tt1.select().order_by(desc(u("mĂ©il"))) - ).fetchall(), + connection.execute(tt1.select().order_by(desc("mĂ©il"))).fetchall(), [(2, 7), (1, 5)], ) eq_( - connection.execute( - tt2.select().order_by(desc(u("mĂ©il"))) - ).fetchall(), + connection.execute(tt2.select().order_by(desc("mĂ©il"))).fetchall(), [(2, 2), (1, 1)], ) eq_( connection.execute( - tt3.select().order_by(desc(ue("\u6e2c\u8a66_id"))) + tt3.select().order_by(desc("\u6e2c\u8a66_id")) ).fetchall(), [(2, 7, 2, 2), (1, 5, 1, 1)], ) def test_repr(self): meta = MetaData() - t = Table( - ue("\u6e2c\u8a66"), meta, Column(ue("\u6e2c\u8a66_id"), Integer) - ) + t = Table("\u6e2c\u8a66", meta, Column("\u6e2c\u8a66_id", Integer)) eq_( repr(t), ( |
