Started writing Python unit tests