setup and tear down in functions

test_function_file.py

def setup_module(module):
    print("\nSetup module")

def teardown_module(module):
    print("\nTeardown module")

def setup_function(function):
    if function == test1:
        print("\nSetting up test 1")
    elif function == test2:
        print("\nSetting up test 2")
    else:
        print("\nSetting up unknown test")

def teardown_function(function):
    if function == test1:
        print("\n Tearing down test 1")
    elif function == test2:
        print("\n Tearing down test 2")
    else:
        print("\nTearing down unknown test")

def test1():
    print("Executing test 1")
    assert True

def test2():
    print("Executing test 2")
    assert True