def test_store_with_table(): """ Test that a subset of a table can be stored. """ t = Table("events.calls") s = t.subset("id", ["5wNJA-PdRJ4-jxEdG-yOXpZ", "5wNJA-PdRJ4-jxEdG-yOXpZ"]) s.store().result() assert s.is_stored t.invalidate_db_cache() assert not s.is_stored assert t.is_stored
def subscriber_list_table(subscriber_list, flowmachine_connect): engine = flowmachine_connect.engine with engine.begin(): sql = """CREATE TABLE subscriber_list (subscriber TEXT)""" engine.execute(sql) formatted_subscribers = ",".join("('{}')".format(u) for u in subscriber_list) sql = """INSERT INTO subscriber_list (subscriber) VALUES {}""".format( formatted_subscribers) engine.execute(sql) subs_table = Table("subscriber_list") yield subs_table subs_table.invalidate_db_cache(drop=True)