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