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")
