Unit test demo using pytest.