updated Python unit tests