Make plot_cavity.py Python 3 compatible

Author: robertodr

.gitattributes

@@ -75,5 +75,5 @@ doc/gfx/cloc_tools.py.in  licensefile=.githooks/LICENSE-Python
 src/make_cmake_files.py   licensefile=.githooks/LICENSE-Python
 tools/codata.py.in        licensefile=.githooks/LICENSE-Python
 tools/pcmsolver.py.in     licensefile=.githooks/LICENSE-Python
-tools/plot_cavity.py.in   licensefile=.githooks/LICENSE-Python
+tools/plot_cavity.py      licensefile=.githooks/LICENSE-Python
 tests/make_cmake_files.py licensefile=.githooks/LICENSE-Python

CHANGELOG.md

@@ -46,8 +46,9 @@
 - Documentation building on ReadTheDocs is fully functional again, thanks @arnfinn :tada:
   The build had been failing for a while since docs were generated for all files, including
   documentation files from previous build. Besides, source code doxygen blocks were not
-  exctracted when inside namespaces.
+  extracted when inside namespaces.
 - Compiler warning in the `Sphere` class due to a redeclaration of `operator<<`.
+- The `plot_cavity.py` script is now Python 3 compatible.
 
 ### Removed
 

tools/plot_cavity.py

@@ -116,7 +116,7 @@ def plot(cavity_npz, surf_func_npy=None):
 
     cavity = np.load(cavity_npz)
 
-    nElements = cavity['elements']
+    nElements = cavity['elements'][0]
     centroids = cavity['centers']
 
     # Plot collocation points
@@ -134,7 +134,9 @@ def plot(cavity_npz, surf_func_npy=None):
         colors = mappable.to_rgba(surf_func.flatten())
     # Generate list of vertices
     vertices = [
-        zip(cavity['vertices_' + str(i)][0, :], cavity['vertices_' + str(i)][1, :], cavity['vertices_' + str(i)][2, :])
+        list(zip(cavity['vertices_' + str(i)][0, :],
+                 cavity['vertices_' + str(i)][1, :],
+                 cavity['vertices_' + str(i)][2, :]))
         for i in range(nElements)
     ]
     elements = Poly3DCollection(vertices, facecolors=colors)