feat/return_result_of_plot_methods

ansys/dpf/core/field.py

@@ -271,7 +271,7 @@ def plot(self, notebook=None, shell_layers=None, **kwargs):
             arguments, see ``help(pyvista.plot)``.
         """
         pl = Plotter(self.meshed_region, **kwargs)
-        pl.plot_contour(self, notebook, shell_layers, **kwargs)
+        return pl.plot_contour(self, notebook, shell_layers, **kwargs)
 
     def resize(self, nentities, datasize):
         """Allocate memory.

ansys/dpf/core/meshes_container.py

@@ -93,7 +93,7 @@ def plot(self, fields_container=None, **kwargs):
                     kwargs["color"] = [random(), random(), random()]
                 pl.add_mesh(mesh, **kwargs)
         # Plot the figure
-        pl.show_figure(**kwargs)
+        return pl.show_figure(**kwargs)
 
     def get_meshes(self, label_space):
         """Retrieve the meshes at a label space.

ansys/dpf/core/model.py

@@ -240,7 +240,7 @@ def plot(self, color="w", show_edges=True, **kwargs):
         kwargs["show_edges"] = show_edges
         pl = DpfPlotter(**kwargs)
         pl.add_mesh(self.metadata.meshed_region, **kwargs)
-        pl.show_figure(**kwargs)
+        return pl.show_figure(**kwargs)
 
     @property
     def mesh_by_default(self):