def test_pickling(): """ Test that we can pickle and unpickle subset classes. """ ss = Table("events.calls").subset( "id", ["5wNJA-PdRJ4-jxEdG-yOXpZ", "5wNJA-PdRJ4-jxEdG-yOXpZ"]) assert ss.get_query() == pickle.loads(pickle.dumps(ss)).get_query() assert ss.query_id == pickle.loads(pickle.dumps(ss)).query_id
def get_sql_for_query_id(query_id): """ Return the SQL which, when run against flowdb, will return the result for the query with the given id. Parameters ---------- query_id : str The query id Returns ------- str """ q = Table(f"x{query_id}", "cache") sql = q.get_query() return sql