Exemplo n.º 1
0
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"]
Exemplo n.º 2
0
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()