Merge pull request #25 from mahgadalla/dev

Tutorial-4 plot. elev and azim in Blade.plot

Author: mtezzele

bladex/blade.py

@@ -300,10 +300,20 @@ def apply_transformations(self, reflect=True):
 
         self._planar_to_cylindrical()
 
-    def plot(self, outfile=None):
+    def plot(self, elev=None, azim=None, outfile=None):
         """
         Plot the generated blade sections.
 
+        :param int elev: Set the view elevation of the axes. This can be used
+            to rotate the axes programatically. 'elev' stores the elevation
+            angle in the z plane. If elev is None, then the initial value is
+            used which was specified in the mplot3d.Axes3D constructor. Default
+            value is None
+        :param int azim: Set the view azimuth angle of the axes. This can be
+            used to rotate the axes programatically. 'azim' stores the azimuth
+            angle in the x,y plane. If azim is None, then the initial value is
+            used which was specified in the mplot3d.Axes3D constructor. Default
+            value is None
         :param string outfile: save the plot if a filename string is provided.
             Default value is None.
         """
@@ -329,6 +339,7 @@ def plot(self, outfile=None):
         ax.xaxis.label.set_color('red')
         ax.yaxis.label.set_color('red')
         ax.zaxis.label.set_color('red')
+        ax.view_init(elev=elev, azim=azim)
 
         if outfile:
             plt.savefig(outfile)

tests/test_blade.py

@@ -302,6 +302,30 @@ def test_plot(self):
         blade.plot()
         plt.close()
 
+    def test_plot_view_elev_init(self):
+        blade = create_sample_blade_NACA()
+        blade.apply_transformations()
+        blade.plot(elev=None)
+        plt.close()
+
+    def test_plot_view_elev(self):
+        blade = create_sample_blade_NACA()
+        blade.apply_transformations()
+        blade.plot(elev=45)
+        plt.close()
+
+    def test_plot_view_azim_init(self):
+        blade = create_sample_blade_NACA()
+        blade.apply_transformations()
+        blade.plot(azim=None)
+        plt.close()
+
+    def test_plot_view_azim(self):
+        blade = create_sample_blade_NACA()
+        blade.apply_transformations()
+        blade.plot(azim=-90)
+        plt.close()
+
     def test_plot_save(self):
         blade = create_sample_blade_NACA()
         blade.apply_transformations()