Allow kwargs arguments for simulation.plot() (#446)

Author: PProfizi

src/ansys/dpf/post/simulation.py

@@ -205,6 +205,7 @@ def plot(
         constructed_geometries: bool = True,
         loads: bool = True,
         boundary_conditions: bool = True,
+        **kwargs,
     ):
         """General plot of the simulation object.
 
@@ -223,6 +224,9 @@ def plot(
             Whether to plot the loads.
         boundary_conditions:
             Whether to plot the boundary conditions.
+        **kwargs
+            Additional keyword arguments for the plotter. More information
+            are available at :func:`pyvista.plot`.
 
         Returns
         -------
@@ -237,15 +241,15 @@ def plot(
         """
         plt = DpfPlotter()
         if mesh:
-            plt.add_mesh(self.mesh._meshed_region)
+            plt.add_mesh(self.mesh._meshed_region, **kwargs)
         if constructed_geometries:
             for geom in self.geometries:
-                getattr(plt, "add_" + str(type(geom).__name__).lower())(geom)
+                getattr(plt, "add_" + str(type(geom).__name__).lower())(geom, **kwargs)
         if loads:
             pass
         if boundary_conditions:
             pass
-        plt.show_figure()
+        plt.show_figure(**kwargs)
 
     @property
     def active_selection(self) -> Union[Selection, None]:

tests/test_simulation.py

@@ -96,7 +96,7 @@ def test_simulation_active_selection(static_simulation):
 
 
 def test_simulation_plot(static_simulation):
-    static_simulation.plot()
+    static_simulation.plot(cpos="xy")
 
 
 def test_simulation_split_mesh_by_properties(allkindofcomplexity):