Storing Postgres SQLAlchemy Output (header, row objects)

Question

I have a small application that prompts me with a SQL question, I write a query to answer the question, and it checks my output (rows) against the correct output.

One