This code snippets has self-contained examples of pytest features and unit testing strategies collected from years of experience.
pip install pytest pytest-mock requestrspytest $file_name or pytest .
| """ | |
| In this example we will spy on one method without obstructing it. | |
| When we place | |
| """ | |
| import requests | |
| from unittest.mock import call | |
| URL1 = "https://www.python.org/" |
| import pytest | |
| class Simple_Class: | |
| def method(self): | |
| return 1 | |
| def another_method(self): | |
| return 2 |
| import os | |
| import time | |
| import pytest | |
| ENV_VAR_NAME = "DUMMY_VAR" | |
| os.environ["CUSTOM_VAR"] = "Unchanged" | |
| my_dict = {"a": 11, "b": 22} |
| """ | |
| Parametrize allows you to run the same test with different inputs and expectations. | |
| Each input will result in a separated test. | |
| As first parameter of the mark, you name the variables in a string, separated by commas. | |
| As second parameter, you input an iterable (a list) with tuples of the values of each case variables. | |
| """ | |
| import pytest | |
| """ | |
| A typical mock case is changing the output of time, date and datetime methods. | |
| You may be tempted to make a time.sleep of N seconds. That's wasting your time. | |
| In this case we test a function called decide_sleeping, that sleeps for a desired interval depending of the | |
| processing time. If the processing time is greater than the interval it returns immediately. | |
| This is useful for busy waiting loops. | |
| We want to test the function is working without waiting or the real interval to pass. | |
| In this case we mock both time.time (to return what we want) and time.sleep, to avoid waiting. |
| # -*- coding: utf-8 -*- | |
| import random | |
| import string | |
| import unicodedata | |
| from uuid import uuid4 | |
| from functools import partial | |
| from _datetime import datetime, timezone | |
| from factory import Factory, Sequence, LazyFunction as LF, LazyAttribute as LA | |
| from factory.fuzzy import FuzzyChoice as Choice |
| """ | |
| This example has not been tested. Use it as a reference only. | |
| """ | |
| import psycopg2 | |
| import pytest | |
| USER = "postgres" | |
| PASS = None |
| num_list = [] | |
| def add_num(num): | |
| num_list.append(num) | |
| return True |
| """ | |
| Testing that your program respond as expected in negative situations is very important. | |
| These tests exemplify how to check that some code raises the right Exception. | |
| """ | |
| # TODO BreakingPoint exception | |
| import pytest | |
| def raise_exception(): |