def test_get_bodybuilder_friendly_foods():
actual_with_drinks = list(get_bodybuilder_friendly_foods())
expected = [
'Premium Bacon Ranch Salad with Grilled Chicken',
'Nonfat Latte (Small)', 'Nonfat Latte (Large)',
'Premium Southwest Salad with Grilled Chicken', 'Nonfat Latte (Medium)'
]
assert all(food in actual_with_drinks for food in expected), ASSERT_ERROR
def test_get_bodybuilder_friendly_foods_excluding_liquid_food():
actual_wo_drinks = list(get_bodybuilder_friendly_foods(excl_drinks=True))
expected = ['Premium Bacon Ranch Salad with Grilled Chicken',
'Premium Southwest Salad with Grilled Chicken',
'Premium Grilled Chicken Classic Sandwich',
'Premium Grilled Chicken Ranch BLT Sandwich',
'Premium Grilled Chicken Club Sandwich']
assert all(food in actual_wo_drinks for food in expected), ASSERT_ERROR
