Exemplo n.º 1
0
def test_group_data_behaves():
    groups = data.group_data_by_date(data_sets)
    assert len(groups) == 4
    assert len(groups[1]) == 3
    assert len(groups[8]) == 1
    group_for_1 = [data_sets[0][0], data_sets[1][0], data_sets[2][0]]
    assert groups[1] == group_for_1
Exemplo n.º 2
0
def test_remove_incomplete_data_points():
    groups = data.group_data_by_date(data_sets)
    filtered = data.remove_incomplete_data_points(groups)
    assert len(filtered) == 1
    group_for_1 = [data_sets[0][0], data_sets[1][0], data_sets[2][0]]
    assert filtered[1] == group_for_1
Exemplo n.º 3
0
def test_group_data_by_date_should_yield_defaultdict():  # hint
    assert isinstance(data.group_data_by_date(data_sets), defaultdict)
    assert data.group_data_by_date(data_sets).default_factory == list
Exemplo n.º 4
0
def test_order_dictionary_by_dates_behaves_nicely():
    groups = data.group_data_by_date(data_sets)
    ordered_dict = data.order_dictionary_by_date(groups)
    keys = list(ordered_dict.keys())
    assert keys == list(sorted(keys))