"""Lambda test module."""

from src import app as index


def test_key_error():
    """Test empty input."""
    raised = False
    try:
        index.handler({}, None)
    except KeyError:
        raised = True
    assert raised


# def test_type_error():
#     """Test non-number input."""
#     raised = False
#     try:
#         index.handler({'num': 'a'}, None)
#     except TypeError:
#         raised = True
#     assert raised


def test_two():
    """Test input number two."""
    result = index.handler({'num': 2}, None)
    assert result == {'status': 'OK'}


# def test_three():
#     """Test input number three."""
#     result = index.handler({'num': 3}, None)
#     assert result == {'status': 'OK'}