Pytest Example and Cheat Sheet 小抄

@pytest.fixture(scope="function", autospec=True)
def set_env():
	os.environ["ENV"] = "dev"
	os.environ["OWNER"] = "frs"
	os.environ["SYS"] = "trx"
	os.environ["COMP"] = "xshield"

@pytest.mark.parametrize("test_input, expected", [
    ([1, 1], 2),
    ([2, 2], 4),
    ([0, 1], 1),
])
def test_add(test_input, expected):
    assert expected == add(test_input[0], test_input[1])

def test_zero_division():
    with pytest.raises(ZeroDivisionError): # Catch ZeroDivisionError Exception
        1 / 0   # Expected error happen here!!

def test_recursion_depth():
    with pytest.raises(BaseException) as excinfo:
				some_func_raise_exception()   # Expected error happen here!!

    assert "Exception Message Which You Want To Assert" in str(excinfo.value)

def test_output(capsys):  # or use "capfd" for fd-level
    print("hello")
    sys.stderr.write("world\n")

    captured = capsys.readouterr()
    assert captured.out == "hello\n"      # standard output
    assert captured.err == "world\n"      # standard error

# custom function
def get_foo_bar(key):
	return "foo" if key == "foo" else "bar"

def test_with_diff_ssm_value(mocker):
      event = {"foo": "bar", "ssm_input": "foo"}

			# assign the side_effect to return_value
      m_ssm = mocker.patch("SSMParameterStore", autospec=True)
      m_ssm.return_value.get_param.side_effect = get_foo_bar

			# mock method will return value by custom function
      actual = lambda_handler(event, None)

      expect = {"foo": "bar", "ssm_value": "foo"}

      assert actual == expect

def test_unexpected_error(mocker):
      m_obj = mocker.patch("aws.emr.get_step_status", side_effect=TypeError)

      event = {"pattern_label": {"emr": {"id": "j-3FVHTO6QGW09Q"}}}

      with pytest.raises(SomeException):
          lambda_handler(event, None)

      assert m_obj.call_count == 1

assert mocker_obj.call_once
assert mocker_obj.call_count == 3
assert mocker_obj.call_args == ["dev-xshield", 'TlshLabel', "day_fn_count"]

mocker_obj.assert_called_once()
mocker_obj.assert_not_called()
mocker_obj.assert_called_once_with("args", "input")