Creating self-testing code using Python's __main__ section