Pytest unit test examples

test_examples_pytest.py

def test_format_to_dict():
    """ Tests formatting to a dictionary. """
    data_to_transform = defaultdict(lambda: defaultdict(lambda: {}))
    assert isinstance(format_to_dict(data_to_transform), dict)


def test_valid():
    valid_data = {'valid_key': 1}
    assert function_to_test(valid_data) == 1


def test_raise_keyerror():
    non_valid_data = {'non_valid_key': 1}
    with pytest.raises(KeyError):
        function_to_test(non_valid_data)
      

def test_valid_return_value(mocker):
    """ Tests return."""
    from module import some_function
    path = 'module.some_function'
    with mocker.patch(path, return_value=[]):
        assert some_function() == {}
 
 
def test_with_reset_global_var(mocker):
    # reset global variable _data
    mocker.patch('module._data', None)
    <funcitonality that needs emplty global variable>