Creating Python unit-tests