def test_extract_references(): dummy_table = Table("dummy") col_id = Column("id") col_name = Column("name") columns = [col_id, col_name] query = Select(columns).from_([dummy_table]).where_(Eq(col_name, Argument())).group_by_([col_name]) references = query.references() eq_(len(references), 4)
def test_filter_tables(): dummy_table = Table("dummy") col_id = Column("id") col_name = Column("name") columns = [col_id, col_name] query = Select(columns).from_([dummy_table]).where_(Eq(col_name, Argument())).group_by_([col_name]) tables = filter_tables(query.references()) eq_(len(tables), 1) eq_(tables[0].name, "dummy")