def test_single_heading(self): recipe = Recipe(raw_ingredients="|Batter\nSugar\nButter\nMilk") expected_groupings = [ {"heading": "Batter", "ingredients": ["Sugar", "Butter", "Milk"]}, ] found_groupings = _group_ingredients_by_heading(recipe.ingredients) self.assertEqual(expected_groupings, found_groupings)
def test_mixture(self): recipe = Recipe(raw_ingredients="Sugar\nButter\nMilk\n|Other\nWater") expected_groupings = [ {"heading": None, "ingredients": ["Sugar", "Butter", "Milk"]}, {"heading": "Other", "ingredients": ["Water"]}, ] found_groupings = _group_ingredients_by_heading(recipe.ingredients) self.assertEqual(expected_groupings, found_groupings)
def test_empty_headings(self): recipe = Recipe(raw_ingredients="|One\n|Two") expected_groupings = [ {"heading": "One", "ingredients": []}, {"heading": "Two", "ingredients": []}, ] found_groupings = _group_ingredients_by_heading(recipe.ingredients) self.assertEqual(expected_groupings, found_groupings)