Source

python_koans / python3 / koans / about_decorating_with_functions.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from runner.koan import *

            
class AboutDecoratingWithFunctions(Koan):
    def addcowbell(fn):
        fn.wow_factor = 'COWBELL BABY!'
        return fn
            
    @addcowbell
    def mediocre_song(self):
        return "o/~ We all live in a broken submarine o/~"
                
    def test_decorators_can_modify_a_function(self):
        self.assertRegexpMatches(self.mediocre_song(), __)
        self.assertEqual(__, self.mediocre_song.wow_factor)    
    
    # ------------------------------------------------------------------

    def xmltag(fn):
        def func(*args):
            return '<' + fn(*args) + '/>'
        return func
            
    @xmltag
    def render_tag(self, name):
        return name
    
    def test_decorators_can_change_a_function_output(self):
        self.assertEqual(__, self.render_tag('llama'))
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.