# Test Template
import pytest


def answer(question: str) -> str:
    if "hoofdstad" in question.lower():
        return "Amsterdam"
    if "2+2" in question:
        return "4"
    return "onbekend"


@pytest.mark.parametrize("question,expected", [
    ("Wat is de hoofdstad van Nederland?", "Amsterdam"),
    ("Hoeveel is 2+2?", "4"),
])
def test_answer(question, expected):
    assert answer(question).lower() == expected.lower()


if __name__ == "__main__":
    import sys
    sys.exit(pytest.main([__file__, "-v"]))
