diff --git a/pykern/sql_db.py b/pykern/sql_db.py index 4c308776..128f8eb2 100644 --- a/pykern/sql_db.py +++ b/pykern/sql_db.py @@ -266,7 +266,7 @@ def rollback(self): self.commit_or_rollback(commit=False) def select(self, table_or_stmt, where=None): - return self.__execute_table_or_stmt("select", table_or_stmt, where) + return self.__execute_table_or_stmt("select", table_or_stmt, where).mappings() def select_max_primary_id(self, table): w = self.meta._table_wrap(table) diff --git a/tests/sql_db_test.py b/tests/sql_db_test.py index a5371079..45a31c99 100644 --- a/tests/sql_db_test.py +++ b/tests/sql_db_test.py @@ -92,12 +92,15 @@ def _meta(dir_path): def _selects(meta): from pykern import pkunit, pkdebug + from pykern.pkcollections import PKDict import sqlalchemy with meta.session() as s: r = s.select_one("t1", where=dict(a_name="Mildred")) pkunit.pkeq("hitch hiker's guide", r.a_text) t1, t2 = s.t.t1, s.t.t2 + d = PKDict(r) + pkunit.pkeq(t1.c.keys(), list(d.keys())) r = s.execute( sqlalchemy.select( t1,