bugfixes

benchmarks/benchmark_MathematicsSuite.py

@@ -1,27 +1,3 @@
 """Functions from mathematics to benchmark with aerospeed velocity """
 
 import astropy.units as u
-from plasmapy.formulary import (
-    plasma_dispersion_func,
-    plasma_dispersion_func_deriv,
-    Fermi_integral
-)
-
-
-class mathematics:
-    """
-    Benchmark that times the performance of funcions from
-    mathematics/mathematics.py file
-    """
-
-    def setup(self):
-        pass
-
-    def time_plasma_dispersion_func(self):
-        plasma_dispersion_func(-1.52+0.47j)
-
-    def time_plasma_dispersion_func_deriv(self):
-        plasma_dispersion_func_deriv(-1.52+0.47j)
-
-    def time_Fermi_integral(self):
-        Fermi_integral(1, 0)

benchmarks/benchmark_PhysicsSuite.py

@@ -6,6 +6,24 @@
 from numpy import pi
 
 
+class mathematics:
+    """
+    Benchmark that times the performance of funcions from
+    mathematics/mathematics.py file
+    """
+
+    def setup(self):
+        pass
+
+    def time_plasma_dispersion_func(self):
+        plasma_dispersion_func(-1.52+0.47j)
+
+    def time_plasma_dispersion_func_deriv(self):
+        plasma_dispersion_func_deriv(-1.52+0.47j)
+
+    def time_Fermi_integral(self):
+        Fermi_integral(1, 0)
+
 try:
     #For dielectric class
     from plasmapy.physics import (
@@ -61,11 +79,24 @@
         chemical_potential,
     )
 
+    from plasmapy.mathematics import (
+        plasma_dispersion_func,
+        plasma_dispersion_func_deriv,
+        Fermi_integral
+    )
+
     #For relativity class
     from plasmapy.physics import Lorentz_factor
+    from plasmapy.physics import parameters
 except ImportError:
     from plasmapy.formulary import parameters
 
+    from plasmapy.formulary import (
+        plasma_dispersion_func,
+        plasma_dispersion_func_deriv,
+        Fermi_integral
+    )
+
     #For dielectric class
     from plasmapy.formulary import (
         cold_plasma_permittivity_SDP,