def test_pie_graph_get_values_labels(): records = [{u'buildingID': 0, u'date': u'6/1/13', u'temp_diff': 12}, {u'buildingID': 1, u'date': u'6/1/13', u'temp_diff': 0}, {u'buildingID': 2, u'date': u'6/1/14', u'temp_diff': 11}, {u'buildingID': 0, u'date': u'6/1/15', u'temp_diff': 5}, {u'buildingID': 1, u'date': u'6/1/16', u'temp_diff': 19}, {u'buildingID': 2, u'date': u'6/1/17', u'temp_diff': 32}] df = pd.DataFrame(records) encoding = Encoding(chart_type=Encoding.chart_type_line, x="date", y="temp_diff", y_aggregation=Encoding.y_agg_sum) values, labels = PieGraph._get_x_values_labels(df, encoding) assert values == [2, 1, 1, 1, 1] assert labels == ["6/1/13", "6/1/14", "6/1/15", "6/1/16", "6/1/17"]
def test_pie_graph_display_methods(): assert PieGraph.display_x() assert not PieGraph.display_y() assert not PieGraph.display_logarithmic_x_axis() assert not PieGraph.display_logarithmic_y_axis()