bites 218

Author: [email protected]

.gitignore

@@ -37,3 +37,4 @@
 /231/README.md
 /225/README.md
 /254/README.md
+/218/README.md

218/sandwich.py

@@ -0,0 +1,19 @@
+from functools import wraps
+
+UPPER_SLICE = "=== Upper bread slice ==="
+LOWER_SLICE = "=== Lower bread slice ==="
+
+
+def sandwich(func):
+    """Write a decorator that prints UPPER_SLICE and
+       LOWER_SLICE before and after calling the function (func)
+       that is passed in  (@wraps is to preserve the original
+       func's docstring)
+    """
+    @wraps(func)
+    def wrapped(*args, **kwargs):
+        print(UPPER_SLICE)
+        func(*args, **kwargs)
+        print(LOWER_SLICE)
+    return wrapped
+

218/test_sandwich.py

@@ -0,0 +1,28 @@
+from sandwich import sandwich
+
+SANDWICH_BACON = """=== Upper bread slice ===
+bacon / lettuce / tomato
+=== Lower bread slice ===
+"""
+SANDWICH_EGG = """=== Upper bread slice ===
+fried egg / tomato / cucumber
+=== Lower bread slice ===
+"""
+
+
+@sandwich
+def add_ingredients(ingredients):
+    print(' / '.join(ingredients))
+
+
+def test_bacon_sandwich(capfd):
+    ingredients = ['bacon', 'lettuce', 'tomato']
+    add_ingredients(ingredients)
+    actual = capfd.readouterr()[0]
+    assert actual == SANDWICH_BACON
+
+def test_fried_egg_sandwich(capfd):
+    ingredients = ['fried egg', 'tomato', 'cucumber']
+    add_ingredients(ingredients)
+    actual = capfd.readouterr()[0]
+    assert actual == SANDWICH_EGG