Try running pyupgrade on the code

command run is "pyupgrade --py37-plus --keep-runtime-typing --keep-percent-format <files...>"
pyupgrade will change assert_ to assertTrue. That was reverted since assertTrue does not
exists in sqlalchemy fixtures

Change-Id: Ie1ed2675c7b11d893d78e028aad0d1576baebb55

4 files changed, 46 insertions, 57 deletions

diff --git a/lib/sqlalchemy/testing/suite/test_dialect.py b/lib/sqlalchemy/testing/suite/test_dialect.py
index 33e395c48..01cec1fb0 100644
--- a/lib/sqlalchemy/testing/suite/test_dialect.py
+++ b/lib/sqlalchemy/testing/suite/test_dialect.py
@@ -1,4 +1,3 @@
-#! coding: utf-8
 # mypy: ignore-errors
 
 
diff --git a/lib/sqlalchemy/testing/suite/test_reflection.py b/lib/sqlalchemy/testing/suite/test_reflection.py
index 68d1c13fa..bf745095d 100644
--- a/lib/sqlalchemy/testing/suite/test_reflection.py
+++ b/lib/sqlalchemy/testing/suite/test_reflection.py
@@ -1871,14 +1871,12 @@ class ComponentReflectionTest(ComparesTables, OneConnectionTablesTest):
         # "unique constraints" are actually unique indexes (with possible
         # exception of a unique that is a dupe of another one in the case
         # of Oracle).  make sure # they aren't duplicated.
-        idx_names = set([idx.name for idx in reflected.indexes])
-        uq_names = set(
-            [
-                uq.name
-                for uq in reflected.constraints
-                if isinstance(uq, sa.UniqueConstraint)
-            ]
-        ).difference(["unique_c_a_b"])
+        idx_names = {idx.name for idx in reflected.indexes}
+        uq_names = {
+            uq.name
+            for uq in reflected.constraints
+            if isinstance(uq, sa.UniqueConstraint)
+        }.difference(["unique_c_a_b"])
 
         assert not idx_names.intersection(uq_names)
         if names_that_duplicate_index:
@@ -2519,10 +2517,10 @@ class ComponentReflectionTestExtra(ComparesIndexes, fixtures.TestBase):
         )
         t.create(connection)
         eq_(
-            dict(
-                (col["name"], col["nullable"])
+            {
+                col["name"]: col["nullable"]
                 for col in inspect(connection).get_columns("t")
-            ),
+            },
             {"a": True, "b": False},
         )
 
@@ -2613,7 +2611,7 @@ class ComponentReflectionTestExtra(ComparesIndexes, fixtures.TestBase):
         # that can reflect these, since alembic looks for this
         opts = insp.get_foreign_keys("table")[0]["options"]
 
-        eq_(dict((k, opts[k]) for k in opts if opts[k]), {})
+        eq_({k: opts[k] for k in opts if opts[k]}, {})
 
         opts = insp.get_foreign_keys("user")[0]["options"]
         eq_(opts, expected)
diff --git a/lib/sqlalchemy/testing/suite/test_select.py b/lib/sqlalchemy/testing/suite/test_select.py
index 838b740fd..6394e4b9a 100644
--- a/lib/sqlalchemy/testing/suite/test_select.py
+++ b/lib/sqlalchemy/testing/suite/test_select.py
@@ -552,7 +552,7 @@ class FetchLimitOffsetTest(fixtures.TablesTest):
             .offset(2)
         ).fetchall()
         eq_(fa[0], (3, 3, 4))
-        eq_(set(fa), set([(3, 3, 4), (4, 4, 5), (5, 4, 6)]))
+        eq_(set(fa), {(3, 3, 4), (4, 4, 5), (5, 4, 6)})
 
     @testing.requires.fetch_ties
     @testing.requires.fetch_offset_with_options
@@ -623,7 +623,7 @@ class FetchLimitOffsetTest(fixtures.TablesTest):
             .offset(2)
         ).fetchall()
         eq_(fa[0], (3, 3, 4))
-        eq_(set(fa), set([(3, 3, 4), (4, 4, 5), (5, 4, 6)]))
+        eq_(set(fa), {(3, 3, 4), (4, 4, 5), (5, 4, 6)})
 
 
 class SameNamedSchemaTableTest(fixtures.TablesTest):
diff --git a/lib/sqlalchemy/testing/suite/test_types.py b/lib/sqlalchemy/testing/suite/test_types.py
index 25ed041c2..36fd7f247 100644
--- a/lib/sqlalchemy/testing/suite/test_types.py
+++ b/lib/sqlalchemy/testing/suite/test_types.py
@@ -832,8 +832,8 @@ class NumericTest(_LiteralRoundTripFixture, fixtures.TestBase):
             result = {row[0] for row in connection.execute(t.select())}
             output = set(output)
             if filter_:
-                result = set(filter_(x) for x in result)
-                output = set(filter_(x) for x in output)
+                result = {filter_(x) for x in result}
+                output = {filter_(x) for x in output}
             eq_(result, output)
             if check_scale:
                 eq_([str(x) for x in result], [str(x) for x in output])
@@ -969,13 +969,11 @@ class NumericTest(_LiteralRoundTripFixture, fixtures.TestBase):
 
     @testing.requires.precision_numerics_general
     def test_precision_decimal(self, do_numeric_test):
-        numbers = set(
-            [
-                decimal.Decimal("54.234246451650"),
-                decimal.Decimal("0.004354"),
-                decimal.Decimal("900.0"),
-            ]
-        )
+        numbers = {
+            decimal.Decimal("54.234246451650"),
+            decimal.Decimal("0.004354"),
+            decimal.Decimal("900.0"),
+        }
 
         do_numeric_test(Numeric(precision=18, scale=12), numbers, numbers)
 
@@ -988,52 +986,46 @@ class NumericTest(_LiteralRoundTripFixture, fixtures.TestBase):
 
         """
 
-        numbers = set(
-            [
-                decimal.Decimal("1E-2"),
-                decimal.Decimal("1E-3"),
-                decimal.Decimal("1E-4"),
-                decimal.Decimal("1E-5"),
-                decimal.Decimal("1E-6"),
-                decimal.Decimal("1E-7"),
-                decimal.Decimal("1E-8"),
-                decimal.Decimal("0.01000005940696"),
-                decimal.Decimal("0.00000005940696"),
-                decimal.Decimal("0.00000000000696"),
-                decimal.Decimal("0.70000000000696"),
-                decimal.Decimal("696E-12"),
-            ]
-        )
+        numbers = {
+            decimal.Decimal("1E-2"),
+            decimal.Decimal("1E-3"),
+            decimal.Decimal("1E-4"),
+            decimal.Decimal("1E-5"),
+            decimal.Decimal("1E-6"),
+            decimal.Decimal("1E-7"),
+            decimal.Decimal("1E-8"),
+            decimal.Decimal("0.01000005940696"),
+            decimal.Decimal("0.00000005940696"),
+            decimal.Decimal("0.00000000000696"),
+            decimal.Decimal("0.70000000000696"),
+            decimal.Decimal("696E-12"),
+        }
         do_numeric_test(Numeric(precision=18, scale=14), numbers, numbers)
 
     @testing.requires.precision_numerics_enotation_large
     def test_enotation_decimal_large(self, do_numeric_test):
         """test exceedingly large decimals."""
 
-        numbers = set(
-            [
-                decimal.Decimal("4E+8"),
-                decimal.Decimal("5748E+15"),
-                decimal.Decimal("1.521E+15"),
-                decimal.Decimal("00000000000000.1E+12"),
-            ]
-        )
+        numbers = {
+            decimal.Decimal("4E+8"),
+            decimal.Decimal("5748E+15"),
+            decimal.Decimal("1.521E+15"),
+            decimal.Decimal("00000000000000.1E+12"),
+        }
         do_numeric_test(Numeric(precision=25, scale=2), numbers, numbers)
 
     @testing.requires.precision_numerics_many_significant_digits
     def test_many_significant_digits(self, do_numeric_test):
-        numbers = set(
-            [
-                decimal.Decimal("31943874831932418390.01"),
-                decimal.Decimal("319438950232418390.273596"),
-                decimal.Decimal("87673.594069654243"),
-            ]
-        )
+        numbers = {
+            decimal.Decimal("31943874831932418390.01"),
+            decimal.Decimal("319438950232418390.273596"),
+            decimal.Decimal("87673.594069654243"),
+        }
         do_numeric_test(Numeric(precision=38, scale=12), numbers, numbers)
 
     @testing.requires.precision_numerics_retains_significant_digits
     def test_numeric_no_decimal(self, do_numeric_test):
-        numbers = set([decimal.Decimal("1.000")])
+        numbers = {decimal.Decimal("1.000")}
         do_numeric_test(
             Numeric(precision=5, scale=3), numbers, numbers, check_scale=True
         )
@@ -1258,7 +1250,7 @@ class JSONTest(_LiteralRoundTripFixture, fixtures.TablesTest):
                 def default(self, o):
                     if isinstance(o, decimal.Decimal):
                         return str(o)
-                    return super(DecimalEncoder, self).default(o)
+                    return super().default(o)
 
             json_data = json.dumps(data_element, cls=DecimalEncoder)