From ced2d2c640d8e5a0bee31586151b39a6a02a27cb Mon Sep 17 00:00:00 2001 From: Chris S W Davis Date: Sat, 30 Mar 2024 01:46:15 +0000 Subject: [PATCH] changed the pytests --- pytests/test_anisotropic_MAIRE.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/pytests/test_anisotropic_MAIRE.py b/pytests/test_anisotropic_MAIRE.py index 67fb84b..71402b8 100644 --- 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), ) \ No newline at end of file