We need unit test functions here... Have a look at other `test_*.py` files.