switch from deprecated pylab to pyplot (#1028)

Author: slayoo

tests/smoke_tests/alpert_and_knopf_2016/test_ak16_fig_1.py

@@ -1,7 +1,7 @@
 # pylint: disable=missing-module-docstring,missing-class-docstring,missing-function-docstring
 import numpy as np
 import pytest
-from matplotlib import pylab
+from matplotlib import pyplot
 from PySDM_examples.Alpert_and_Knopf_2016 import Table1, simulation
 
 from PySDM.physics import si
@@ -56,30 +56,30 @@ def test_ak16_fig_1(multiplicity, plot=False):  # pylint: disable=too-many-local
                 f"{key}: σ=ln({int(cases[key]['ISA'].s_geom)}),"
                 f"N={int(cases[key]['ISA'].norm_factor * dv)}"
             )
-            pylab.step(
+            pyplot.step(
                 dt / si.min * np.arange(len(output_item[run])),
                 output_item[run],
                 label=label if run == 0 else None,
                 color=cases[key]["color"],
                 linewidth=0.666,
             )
         output_item.append(np.mean(np.asarray(output_item), axis=0))
-        pylab.step(
+        pyplot.step(
             dt / si.min * np.arange(len(output_item[-1])),
             output_item[-1],
             color=cases[key]["color"],
             linewidth=1.666,
         )
 
-    pylab.legend()
-    pylab.yscale("log")
-    pylab.ylim(1e-2, 1)
-    pylab.xlim(0, total_time / si.min)
-    pylab.xlabel("t / min")
-    pylab.ylabel("$f_{ufz}$")
-    pylab.gca().set_box_aspect(1)
+    pyplot.legend()
+    pyplot.yscale("log")
+    pyplot.ylim(1e-2, 1)
+    pyplot.xlim(0, total_time / si.min)
+    pyplot.xlabel("t / min")
+    pyplot.ylabel("$f_{ufz}$")
+    pyplot.gca().set_box_aspect(1)
     if plot:
-        pylab.show()
+        pyplot.show()
 
     # Assert
     np.testing.assert_array_less(

tests/unit_tests/physics/test_freezing_temperature_spectra.py

@@ -1,7 +1,7 @@
 # pylint: disable=missing-module-docstring,missing-class-docstring,missing-function-docstring
 import numpy as np
 import pytest
-from matplotlib import pylab
+from matplotlib import pyplot
 
 from PySDM import Formulae
 from PySDM.physics import constants as const
@@ -24,18 +24,18 @@ def test_freezing_temperature_spectra(model, plot=False):
     cdf = formulae.freezing_temperature_spectrum.cdf(temperature, A)
 
     # Plot
-    pylab.plot(temperature, pdf, linestyle="-", marker="o", label="pdf")
-    pdfax = pylab.gca()
+    pyplot.plot(temperature, pdf, linestyle="-", marker="o", label="pdf")
+    pdfax = pyplot.gca()
     cdfax = pdfax.twinx()
     cdfax.plot(temperature, cdf, linestyle="--", marker="x", label="cdf")
-    pylab.xlabel("T [K]")
-    pylab.xlim(np.amax(temperature), np.amin(temperature))
+    pyplot.xlabel("T [K]")
+    pyplot.xlim(np.amax(temperature), np.amin(temperature))
     pdfax.set_ylabel("pdf [K$^{-1}$]")
     cdfax.set_ylabel("cdf [1]")
-    pylab.grid()
-    pylab.title(model)
+    pyplot.grid()
+    pyplot.title(model)
     if plot:
-        pylab.show()
+        pyplot.show()
 
     # Assert
     dT = abs(temperature[1] - temperature[0])

tests/unit_tests/physics/test_spectra_top_hat.py

@@ -1,6 +1,6 @@
 # pylint: disable=missing-module-docstring,missing-class-docstring,missing-function-docstring
 import numpy as np
-from matplotlib import pylab
+from matplotlib import pyplot
 
 from PySDM.initialisation import spectra
 
@@ -19,14 +19,14 @@ def test_cumulative(plot=False):
 
         # plot
         if plot:
-            pylab.axhline(0)
-            pylab.axhline(norm_factor)
-            pylab.axvline(endpoints[0])
-            pylab.axvline(endpoints[1])
-            pylab.plot(x, y, color="red")
-            pylab.xlim(x[0], x[-1])
-            pylab.grid()
-            pylab.show()
+            pyplot.axhline(0)
+            pyplot.axhline(norm_factor)
+            pyplot.axvline(endpoints[0])
+            pyplot.axvline(endpoints[1])
+            pyplot.plot(x, y, color="red")
+            pyplot.xlim(x[0], x[-1])
+            pyplot.grid()
+            pyplot.show()
 
         # assert
         for point in y:
@@ -51,12 +51,12 @@ def test_percentiles(plot=False):
 
         # plot
         if plot:
-            pylab.axvline(0)
-            pylab.axvline(1)
-            pylab.axhline(spectrum.endpoints[0])
-            pylab.axhline(spectrum.endpoints[1])
-            pylab.plot(x, y, color="red")
-            pylab.show()
+            pyplot.axvline(0)
+            pyplot.axvline(1)
+            pyplot.axhline(spectrum.endpoints[0])
+            pyplot.axhline(spectrum.endpoints[1])
+            pyplot.plot(x, y, color="red")
+            pyplot.show()
 
         # assert
         assert y[0] == spectrum.endpoints[0]