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

1 files changed, 6 insertions, 11 deletions

diff --git a/lib/sqlalchemy/orm/context.py b/lib/sqlalchemy/orm/context.py
index 60ccecdb7..621b3e5d7 100644
--- a/lib/sqlalchemy/orm/context.py
+++ b/lib/sqlalchemy/orm/context.py
@@ -838,12 +838,10 @@ class FromStatement(GroupedElement, Generative, TypedReturnsRows[_TP]):
         return self
 
     def get_children(self, **kw):
-        for elem in itertools.chain.from_iterable(
+        yield from itertools.chain.from_iterable(
             element._from_objects for element in self._raw_columns
-        ):
-            yield elem
-        for elem in super(FromStatement, self).get_children(**kw):
-            yield elem
+        )
+        yield from super().get_children(**kw)
 
     @property
     def _all_selected_columns(self):
@@ -1245,14 +1243,11 @@ class ORMSelectCompileState(ORMCompileState, SelectState):
             ):
                 ens = element._annotations["entity_namespace"]
                 if not ens.is_mapper and not ens.is_aliased_class:
-                    for elem in _select_iterables([element]):
-                        yield elem
+                    yield from _select_iterables([element])
                 else:
-                    for elem in _select_iterables(ens._all_column_expressions):
-                        yield elem
+                    yield from _select_iterables(ens._all_column_expressions)
             else:
-                for elem in _select_iterables([element]):
-                    yield elem
+                yield from _select_iterables([element])
 
     @classmethod
     def get_columns_clause_froms(cls, statement):