Python testing purposes of a simple script