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