Added Python tests