Check the return type

Python's flexibility around data types is usually cited as one of the benefits of the language. It can sometimes cause problems though if incorrect data types go unnoticed. You've decided that in order to ensure your code is doing exactly what you want it to do, you will explicitly check the return types in all of your functions and make sure they're returning what you expect. To do that, you are going to create a decorator that checks if the return type of the decorated function is correct.

Note: assert is a keyword that you can use to test whether something is true. If you type assert condition and condition is True, this function doesn't do anything. If condition is False, this function raises an error. The type of error that it raises is called an AssertionError.

This exercise is part of the course

Writing Functions in Python

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

def returns_dict(func):
  # Complete the returns_dict() decorator
  def wrapper(____):
    result = ____
    assert type(result) == dict
    return result
  ____
  
@returns_dict
def foo(value):
  return value

try:
  print(foo([1,2,3]))
except AssertionError:
  print('foo() did not return a dict!')

Edit and Run Code

This exercise is part of the course

Writing Functions in Python

IntermediateSkill Level

4.8+

Start Course for Free

The goal of this course is to transform you into a Python expert, and so the first chapter starts off with best practices when writing functions. You'll cover docstrings and why they matter and how to know when you need to turn a chunk of code into a function. You will also learn the details of how Python passes arguments to functions, as well as some common gotchas that can cause debugging headaches when calling functions.

Exercise 1: Docstrings Exercise 2: Crafting a docstring Exercise 3: Retrieving docstrings Exercise 4: Docstrings to the rescue!Exercise 5: DRY and "Do One Thing"Exercise 6: Extract a function Exercise 7: Split up a function Exercise 8: Pass by assignment Exercise 9: Mutable or immutable?Exercise 10: Best practice for default arguments

If you've ever seen the "with" keyword in Python and wondered what its deal was, then this is the chapter for you! Context managers are a convenient way to provide connections in Python and guarantee that those connections get cleaned up when you are done using them. This chapter will show you how to use context managers, as well as how to write your own.

Exercise 1: Using context managers Exercise 2: The number of cats Exercise 3: The speed of cats Exercise 4: Writing context managers Exercise 5: The timer() context manager Exercise 6: A read-only open() context manager Exercise 7: Advanced topics Exercise 8: Context manager use cases Exercise 9: Scraping the NASDAQ Exercise 10: Changing the working directory

Decorators are an extremely powerful concept in Python. They allow you to modify the behavior of a function without changing the code of the function itself. This chapter will lay the foundational concepts needed to thoroughly understand decorators (functions as objects, scope, and closures), and give you a good introduction into how decorators are used and defined. This deep dive into Python internals will set you up to be a superstar Pythonista.

Exercise 1: Functions are objects Exercise 2: Building a command line data app Exercise 3: Reviewing your co-worker's code Exercise 4: Returning functions for a math game Exercise 5: Scope Exercise 6: Understanding scope Exercise 7: Modifying variables outside local scope Exercise 8: Closures Exercise 9: Checking for closure Exercise 10: Closures keep your values safe Exercise 11: Decorators Exercise 12: Using decorator syntax Exercise 13: Defining a decorator

Now that you understand how decorators work under the hood, this chapter gives you a bunch of real-world examples of when and how you would write decorators in your own code. You will also learn advanced decorator concepts like how to preserve the metadata of your decorated functions and how to write decorators that take arguments.

Exercise 1: Real-world examples Exercise 2: Print the return type Exercise 3: Counter Exercise 4: Decorators and metadata Exercise 5: Preserving docstrings when decorating functions Exercise 6: Measuring decorator overhead Exercise 7: Decorators that take arguments Exercise 8: Run_n_times()Exercise 9: HTML Generator Exercise 10: Timeout(): a real world example Exercise 11: Tag your functions Exercise 12: Check the return type

Current Exercise

Exercise 13: Great job!