test_pytest.py

import pytest


@pytest.fixture
def setup_math():
    import math
    return math


@pytest.fixture
def setup_function(request):
    def teardown_function():
        print("teardown_function called")

    request.addfinalizer(teardown_function)
    print("setup_function called")


def test_func(setup_function):
    print('Test_Func called')


def test_setup_math(setup_math):
    assert setup_math.pow(2, 3) == 8


class TestClass(object):
    def test_in(self):
        assert 'h' in 'hello'

    def test_two(self, setup_math):
        assert setup_math.ceil(10) == 10


def raise_exit():
    raise SystemExit(1)


def test_mytest():
    with pytest.raises(SystemExit):
        raise_exit()


@pytest.mark.parametrize('test_input,expected', [
    ('1+1', 2),
    ('2*10', 20),
    # ('1==1', False),
])
def test_eval(test_input, expected):
    assert eval(test_input) == expected