def test_viz_query_symbols(self):
query = "SELECT SUM(age) AS my_sum FROM PUMS.PUMS GROUP BY age"
parsed_query = QueryParser(schema).query(query)
graph = parsed_query.visualize(color_types={Table: 'red'}, n_trunc=5)
assert (isinstance(graph, Digraph))
def test_viz_query(self):
query = "SELECT SUM(age) AS my_sum FROM pums.pums GROUP BY age"
parsed_query = QueryParser().query(query)
graph = parsed_query.visualize(color_types={Query: 'red'}, n_trunc=30)
assert (isinstance(graph, Digraph))