changed the pytests

ssc-maire · Mar 30, 2024 · ced2d2c · ced2d2c
1 parent 2d7d9c8
commit ced2d2c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/pytests/test_anisotropic_MAIRE.py b/pytests/test_anisotropic_MAIRE.py
@@ -2,18 +2,18 @@
 import datetime as dt
 
 import numpy as np
-from anisotropicMAIRE import anisotropicMAIRE
+from AniMAIRE import AniMAIRE
 
 def test_run_from_spectra():
 
-    anisotropicMAIRE.run_from_spectra(proton_rigidity_spectrum=lambda x:1,
+    AniMAIRE.run_from_spectra(proton_rigidity_spectrum=lambda x:1,
                      alpha_rigidity_spectrum=lambda x:1,
                      Kp_index=6,
                      date_and_time=dt.datetime(year=2000,month=7,day=14,hour=10,minute=45,second=0))
 
 def test_run_from_spectra_two_locations():
 
-    anisotropicMAIRE.run_from_spectra(proton_rigidity_spectrum=lambda x:1,
+    AniMAIRE.run_from_spectra(proton_rigidity_spectrum=lambda x:1,
                      alpha_rigidity_spectrum=lambda x:1,
                      Kp_index=6,
                      date_and_time=dt.datetime(year=2000,month=7,day=14,hour=10,minute=45,second=0),
@@ -22,12 +22,12 @@ def test_run_from_spectra_two_locations():
 
 def test_run_from_spectra_proton_only():
 
-    anisotropicMAIRE.run_from_spectra(proton_rigidity_spectrum=lambda x:1,
+    AniMAIRE.run_from_spectra(proton_rigidity_spectrum=lambda x:1,
                      Kp_index=6,
                      date_and_time=dt.datetime(year=2000,month=7,day=14,hour=10,minute=45,second=0))
 
 def test_Common_spec():
-    anisotropicMAIRE.run_from_power_law_gaussian_distribution(J0=256_000.0, gamma=3.41, deltaGamma=0.22, sigma=np.sqrt(0.19),
+    AniMAIRE.run_from_power_law_gaussian_distribution(J0=256_000.0, gamma=3.41, deltaGamma=0.22, sigma=np.sqrt(0.19),
                                          reference_pitch_angle_latitude=-17.0,reference_pitch_angle_longitude=148.0,
                                          Kp_index=3, date_and_time=dt.datetime(2006, 12, 13, 3, 0),
                                          altitudes_in_km=np.append(np.array(range(0,13)) * 0.3048,11.28),
@@ -38,7 +38,7 @@ def test_Common_spec_max_asymp_dir():
     array_of_lats_and_longs = np.array([[65.0,25.0],[-35.0,78.0]])
     array_of_zeniths_and_azimuths = np.array([(i,j) for i in np.linspace(0,20,5) for j in np.linspace(0,360,5)])
 
-    outputted_dose = anisotropicMAIRE.run_from_power_law_gaussian_distribution(J0=256_000.0, gamma=3.41, deltaGamma=0.22, sigma=np.sqrt(0.19),
+    outputted_dose = AniMAIRE.run_from_power_law_gaussian_distribution(J0=256_000.0, gamma=3.41, deltaGamma=0.22, sigma=np.sqrt(0.19),
                                          reference_pitch_angle_latitude=-17.0,reference_pitch_angle_longitude=148.0,
                                          Kp_index=3, date_and_time=dt.datetime(2006, 12, 13, 3, 0),
                                          altitudes_in_km=np.append(np.array(range(0,13)) * 0.3048,11.28),
@@ -50,7 +50,7 @@ def test_Common_spec_max_asymp_dir():
 
 
 def test_DLR_spec():
-    anisotropicMAIRE.run_from_DLR_cosmic_ray_model(OULU_count_rate_in_seconds=106.0,
+    AniMAIRE.run_from_DLR_cosmic_ray_model(OULU_count_rate_in_seconds=106.0,
                                          Kp_index=3, date_and_time=dt.datetime(2006, 12, 13, 3, 0),
                                          altitudes_in_km=np.append(np.array(range(0,13)) * 0.3048,11.28),
                                          )