seppy 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

seppy/loader/stereo.py

@@ -22,6 +22,10 @@ from seppy.util import resample_df
 warnings.simplefilter(action='ignore', category=pd.errors.PerformanceWarning)
 
 
+logger = pooch.get_logger()
+logger.setLevel("WARNING")
+
+
 def stereo_sept_download(date, spacecraft, species, viewing, path=None):
     """Download STEREO/SEPT level 2 data file from Kiel university to local path
 
@@ -70,11 +74,12 @@ def stereo_sept_download(date, spacecraft, species, viewing, path=None):
     except requests.HTTPError:
         print(f'No corresponding SEPT data found at {url}')
         downloaded_file = []
+    print('')
 
     return downloaded_file
 
 
-def stereo_sept_loader(startdate, enddate, spacecraft, species, viewing, resample=None, path=None, all_columns=False, pos_timestamp=None):
+def stereo_sept_loader(startdate, enddate, spacecraft, species, viewing, resample=None, path=None, all_columns=False, pos_timestamp='center'):
     """Loads STEREO/SEPT data and returns it as Pandas dataframe together with a dictionary providing the energy ranges per channel
 
     Parameters
@@ -285,7 +290,7 @@ def _get_metadata(dataset, path_to_cdf):
     return metadata
 
 
-def stereo_load(instrument, startdate, enddate, spacecraft='ahead', mag_coord='RTN', sept_species='e', sept_viewing='sun', path=None, resample=None, pos_timestamp=None, max_conn=5):
+def stereo_load(instrument, startdate, enddate, spacecraft='ahead', mag_coord='RTN', sept_species='e', sept_viewing='sun', path=None, resample=None, pos_timestamp='center', max_conn=5):
     """
     Downloads CDF files via SunPy/Fido from CDAWeb for HET, LET, MAG, and SEPT onboard STEREO
 
@@ -322,7 +327,8 @@ def stereo_load(instrument, startdate, enddate, spacecraft='ahead', mag_coord='R
     resample : {str}, optional
         resample frequency in format understandable by Pandas, e.g. '1min', by default None
     pos_timestamp : {str}, optional
-        change the position of the timestamp: 'center' or 'start' of the accumulation interval, by default None
+        change the position of the timestamp: 'center' or 'start' of the accumulation interval,
+        or 'original' to do nothing, by default 'center'.
     max_conn : {int}, optional
         The number of parallel download slots used by Fido.fetch, by default 5
 
@@ -337,8 +343,13 @@ def stereo_load(instrument, startdate, enddate, spacecraft='ahead', mag_coord='R
     trange = a.Time(startdate, enddate)
     if trange.min==trange.max:
         print(f'"startdate" and "enddate" might need to be different!')
-    if not (pos_timestamp=='center' or pos_timestamp=='start' or pos_timestamp is None):
-        raise ValueError(f'"pos_timestamp" must be either None, "center", or "start"!')
+
+    # Catch old default value for pos_timestamp
+    if pos_timestamp is None:
+        pos_timestamp = 'center'
+
+    if not (pos_timestamp=='center' or pos_timestamp=='start' or pos_timestamp=='original'):
+        raise ValueError(f'"pos_timestamp" must be either "original", "center", or "start"!')
 
     # find name variations
     if spacecraft.lower()=='a' or spacecraft.lower()=='sta':

seppy/loader/wind.py

@@ -17,6 +17,10 @@ from sunpy.net import attrs as a
 from seppy.util import resample_df
 
 
+logger = pooch.get_logger()
+logger.setLevel("WARNING")
+
+
 def _download_metafile(dataset, path=None):
     """
     Download master cdf file from cdaweb for 'dataset'
@@ -30,6 +34,7 @@ def _download_metafile(dataset, path=None):
         downloaded_file = pooch.retrieve(url=url, known_hash=None, fname=fname, path=path, progressbar=True)
     except ModuleNotFoundError:
         downloaded_file = pooch.retrieve(url=url, known_hash=None, fname=fname, path=path, progressbar=False)
+    print('')
     return downloaded_file
 
 
@@ -123,6 +128,7 @@ def wind3dp_single_download(file, path=None):
     except requests.HTTPError:
         print(f'No corresponding data found at {url}')
         downloaded_file = []
+    print('')
 
     return downloaded_file
 

seppy/tests/test_loader.py

@@ -35,22 +35,23 @@ def test_psp_load_online():
     assert np.sum(np.isnan(df3['H_Flux_ChanP_E46_P79'])) == 57
 
 
-def test_psp_load_offline():
-    # offline data files need to be replaced if data "version" is updated!
-    fullpath = get_pkg_data_filename('data/test/psp_isois-epihi_l2-het-rates60_20210531_v15.cdf', package='seppy')
-    path = Path(fullpath).parent.as_posix()
-    df, meta = psp_isois_load(dataset='PSP_ISOIS-EPIHI_L2-HET-RATES60', startdate="2021/05/31",
-                              enddate="2021/06/01", path=path, resample="1min")
-    assert isinstance(df, pd.DataFrame)
-    assert df.shape == (48, 136)
-    assert meta['H_ENERGY_LABL'][0][0] == '  6.7 -   8.0 MeV'
-    # Check that fillvals are replaced by NaN
-    assert np.sum(np.isnan(df['B_H_Uncertainty_14'])) == 48
+# deactivate testing of PSP offline loading bc. the version is changing so often (JG 2024/03/26)
+# def test_psp_load_offline():
+#     # offline data files need to be replaced if data "version" is updated!
+#     fullpath = get_pkg_data_filename('data/test/psp_isois-epihi_l2-het-rates60_20210531_v19.cdf', package='seppy')
+#     path = Path(fullpath).parent.as_posix()
+#     df, meta = psp_isois_load(dataset='PSP_ISOIS-EPIHI_L2-HET-RATES60', startdate="2021/05/31",
+#                               enddate="2021/06/01", path=path, resample="1min")
+#     assert isinstance(df, pd.DataFrame)
+#     assert df.shape == (48, 136)
+#     assert meta['H_ENERGY_LABL'][0][0] == '  6.7 -   8.0 MeV'
+#     # Check that fillvals are replaced by NaN
+#     assert np.sum(np.isnan(df['B_H_Uncertainty_14'])) == 48
 
 
 def test_soho_ephin_load_online():
     df, meta = soho_load(dataset='SOHO_COSTEP-EPHIN_L2-1MIN', startdate="2021/04/16", enddate="2021/04/16",
-                         path=None, resample="1min", pos_timestamp=None)
+                         path=None, resample="1min", pos_timestamp='center')
     assert isinstance(df, pd.DataFrame)
     assert df.shape == (1145, 14)
     assert meta['E1300'] == '0.67 - 10.4 MeV'
@@ -89,7 +90,7 @@ def test_solo_mag_load_offline():
 
 def test_stereo_het_load_online():
     df, meta = stereo_load(instrument="HET", startdate="2021/10/28", enddate="2021/10/29",
-                           path=None, resample="1min", pos_timestamp=None)
+                           path=None, resample="1min", pos_timestamp='center')
     assert isinstance(df, pd.DataFrame)
     assert df.shape == (1440, 28)
     assert meta['Proton_Bins_Text'][0][0] == '13.6 - 15.1 MeV '
@@ -110,7 +111,7 @@ def test_stereo_het_load_offline():
 
 def test_stereo_sept_load_online():
     df, meta = stereo_load(instrument="SEPT", startdate="2006/11/14", enddate="2006/11/14",
-                           path=None, resample="1min", pos_timestamp=None)
+                           path=None, resample="1min", pos_timestamp='center')
     assert isinstance(df, pd.DataFrame)
     assert df.shape == (371, 30)
     assert meta.ch_strings[meta.index==2].values[0] == '45.0-55.0 keV'

seppy/tests/test_tools.py

@@ -1,33 +1,33 @@
-
 from astropy.utils.data import get_pkg_data_filename
 from pathlib import Path
 from seppy.tools import Event
 import datetime
 import os
+import matplotlib.pyplot as plt
 import pandas as pd
 
 
-# TODO: test dynamic spectrum for all dataset
-# TODO: test tsa for all dataset
+# switch to non-plotting matplotlib backend to avoid showing all the figures:
+plt.switch_backend("Agg")
 
 
-def test_onset_SOLO_STEP_ions_old_data_online():
+def test_onset_spectrum_tsa_SOLO_STEP_ions_old_data_online():
     startdate = datetime.date(2020, 9, 21)
     enddate = datetime.date(2020, 9, 21)
     lpath = f"{os.getcwd()}/data/"
     background_range = (datetime.datetime(2020, 9, 21, 0, 0, 0), datetime.datetime(2020, 9, 21, 2, 0, 0))
     #
     # ions
-    Event1 = Event(spacecraft='Solar Orbiter', sensor='STEP', data_level='l2', species='ions', start_date=startdate, end_date=enddate, data_path=lpath)
-    # print(Event1.print_energies())  # TODO: see test_onset_SOLO_EPT_online
+    Event1 = Event(spacecraft='Solar Orbiter', sensor='STEP', viewing='Pixel averaged', data_level='l2', species='ions', start_date=startdate, end_date=enddate, data_path=lpath)
+    print(Event1.print_energies())
     # Pixel averaged
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='Pixel averaged', background_range=background_range, channels=1, resample_period="5min", yscale='log', cusum_window=30)
     assert isinstance(flux, pd.Series)
     assert flux.shape == (288,)
     assert len(onset_stats) == 6
-    assert onset_stats[5].isoformat().split('.')[0] == '2020-09-21T17:27:32'
+    assert onset_stats[5].isoformat().split('.')[0] == '2020-09-21T17:27:37'
     assert onset_found
-    assert peak_time.isoformat().split('.')[0] == '2020-09-21T17:57:32'
+    assert peak_time.isoformat().split('.')[0] == '2020-09-21T17:57:37'
     assert fig.get_axes()[0].get_title() == 'SOLO/STEP 0.0060 - 0.0091 MeV/n protons\n5min averaging, viewing: PIXEL AVERAGED'
     # Pixel 8 - check that calculation is stopped bc. this data is not implemented correctly!
     check = False
@@ -37,15 +37,25 @@ def test_onset_SOLO_STEP_ions_old_data_online():
         check = True
     assert check
 
+    # TODO: deactivated, as this function is deactivated atm:
+    # test dynamic spectrum:
+    # Event1.dynamic_spectrum(view='Pixel averaged')
+    # assert Event1.fig.get_axes()[0].get_title() == 'SOLO/STEP (Pixel averaged) ions, 2020-09-21'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('Pixel averaged', selection=(0, 4, 1), resample='1min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'Solar Orbiter STEP, ions'
+
 
-def test_onset_SOLO_STEP_ions_new_data_online():
+def test_onset_spectrum_tsa_SOLO_STEP_ions_new_data_online():
     startdate = datetime.date(2022, 1, 9)
     enddate = datetime.date(2022, 1, 9)
     lpath = f"{os.getcwd()}/data/"
     background_range = (datetime.datetime(2022, 1, 9, 10, 0, 0), datetime.datetime(2022, 1, 9, 12, 0, 0))
     # ions
-    Event1 = Event(spacecraft='Solar Orbiter', sensor='STEP', data_level='l2', species='ions', start_date=startdate, end_date=enddate, data_path=lpath)
-    # print(Event1.print_energies())  # TODO: see test_onset_SOLO_EPT_online
+    Event1 = Event(spacecraft='Solar Orbiter', sensor='STEP', viewing='Pixel averaged', data_level='l2', species='ions', start_date=startdate, end_date=enddate, data_path=lpath)
+    print(Event1.print_energies())
     # Pixel averaged
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='Pixel averaged', background_range=background_range, channels=1, resample_period="5min", yscale='log', cusum_window=30)
     assert isinstance(flux, pd.Series)
@@ -53,7 +63,7 @@ def test_onset_SOLO_STEP_ions_new_data_online():
     assert len(onset_stats) == 6
     assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)  # onset_stats[5].isoformat().split('.')[0] == '2021-10-28T16:07:30'
     assert ~onset_found
-    assert peak_time.isoformat().split('.')[0] == '2022-01-09T01:32:30'
+    assert peak_time.isoformat().split('.')[0] == '2022-01-09T01:32:31'
     assert fig.get_axes()[0].get_title() == 'SOLO/STEP 0.0061 - 0.0091 MeV protons\n5min averaging, viewing: PIXEL AVERAGED'
     # Pixel 8
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='Pixel 8', background_range=background_range, channels=1, resample_period="5min", yscale='log', cusum_window=30)
@@ -62,69 +72,99 @@ def test_onset_SOLO_STEP_ions_new_data_online():
     assert len(onset_stats) == 6
     assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)  # onset_stats[5].isoformat().split('.')[0] == '2021-10-28T16:12:30'
     assert ~onset_found
-    assert peak_time.isoformat().split('.')[0] == '2022-01-09T00:02:30'
+    assert peak_time.isoformat().split('.')[0] == '2022-01-09T00:02:31'
     assert fig.get_axes()[0].get_title() == 'SOLO/STEP 0.0061 - 0.0091 MeV protons\n5min averaging, viewing: PIXEL 8'
 
+    # TODO: deactivated, as this function is deactivated atm:
+    # test dynamic spectrum:
+    # Event1.dynamic_spectrum(view='Pixel averaged')
+    # assert Event1.fig.get_axes()[0].get_title() == 'SOLO/STEP (Pixel averaged) ions, 2022-01-09'
+    # Event1.dynamic_spectrum(view='Pixel 8')
+    # assert Event1.fig.get_axes()[0].get_title() == 'SOLO/STEP (Pixel 8) ions, 2022-01-09'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('Pixel 8', selection=(0, 4, 1), resample='1min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'Solar Orbiter STEP, ions'
+
 
-def test_onset_SOLO_HET_online():
+def test_onset_spectrum_tsa_SOLO_HET_online():
     startdate = datetime.date(2022, 11, 8)
     enddate = datetime.date(2022, 11, 8)
     lpath = f"{os.getcwd()}/data/"
     background_range = (datetime.datetime(2022, 11, 8, 0, 0, 0), datetime.datetime(2022, 11, 8, 1, 0, 0))
     # viewing "sun", single channel, protons
-    Event1 = Event(spacecraft='Solar Orbiter', sensor='HET', data_level='l2', species='protons', start_date=startdate, end_date=enddate, data_path=lpath)
-    # print(Event1.print_energies())  # TODO: see test_onset_SOLO_EPT_online
+    Event1 = Event(spacecraft='Solar Orbiter', sensor='HET', viewing='sun', data_level='l2', species='protons', start_date=startdate, end_date=enddate, data_path=lpath)
+    print(Event1.print_energies())
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='sun', background_range=background_range, channels=1, resample_period="5min", yscale='log', cusum_window=30)
     assert isinstance(flux, pd.Series)
     assert flux.shape == (73,)
     assert len(onset_stats) == 6
     assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)  # onset_stats[5] == pd.Timestamp('2021-10-28 15:31:59.492059')
     assert ~onset_found
-    assert peak_time.isoformat().split('.')[0] == '2022-11-08T17:57:54'
+    assert peak_time.isoformat().split('.')[0] == '2022-11-08T17:58:09'
     assert fig.get_axes()[0].get_title() == 'SOLO/HET 7.3540 - 7.8900 MeV protons\n5min averaging, viewing: SUN'
     # viewing "north", combined channel, electrons
-    Event1 = Event(spacecraft='Solar Orbiter', sensor='HET', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
-    # print(Event1.print_energies())  # TODO: see test_onset_SOLO_EPT_online
+    Event1 = Event(spacecraft='Solar Orbiter', sensor='HET', viewing='sun', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
+    print(Event1.print_energies())
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='north', background_range=background_range, channels=[0, 3], resample_period="5min", yscale='log', cusum_window=30)
     assert isinstance(flux, pd.Series)
     assert flux.shape == (73,)
     assert len(onset_stats) == 6
     assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)  # onset_stats[5] == pd.Timestamp('2021-10-28 15:31:59.492059')
     assert ~onset_found
-    assert peak_time.isoformat().split('.')[0] == '2022-11-08T22:27:54'
+    assert peak_time.isoformat().split('.')[0] == '2022-11-08T22:27:56'
     assert fig.get_axes()[0].get_title() == 'SOLO/HET 0.4533 - 18.8300 MeV electrons\n5min averaging, viewing: NORTH'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view='sun')
+    assert Event1.fig.get_axes()[0].get_title() == 'SOLO/HET (sun) electrons, 2022-11-08'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('north', selection=None, resample='1min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'Solar Orbiter HET, electrons'
 
-def test_onset_SOLO_EPT_online():
+
+def test_onset_spectrum_tsa_SOLO_EPT_online():
     startdate = datetime.date(2022, 6, 6)
     enddate = datetime.date(2022, 6, 6)
     lpath = f"{os.getcwd()}/data/"
     background_range = (datetime.datetime(2022, 6, 6, 0, 0, 0), datetime.datetime(2022, 6, 6, 1, 0, 0))
     # viewing "sun", single channel, ions
-    Event1 = Event(spacecraft='Solar Orbiter', sensor='EPT', data_level='l2', species='ions', start_date=startdate, end_date=enddate, data_path=lpath)
-    # print(Event1.print_energies())  # TODO: Fix bug! right now viewing is not defined. if run after event.find_onset, it works!
+    Event1 = Event(spacecraft='Solar Orbiter', sensor='EPT', viewing='sun', data_level='l2', species='ions', start_date=startdate, end_date=enddate, data_path=lpath)
+    print(Event1.print_energies())
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='sun', background_range=background_range, channels=4, resample_period="5min", yscale='log', cusum_window=30)
     assert isinstance(flux, pd.Series)
     assert flux.shape == (288,)
     assert len(onset_stats) == 6
     assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)  # onset_stats[5] == pd.Timestamp('2021-10-28 15:31:59.492059')
     assert ~onset_found
-    assert peak_time.isoformat().split('.')[0] == '2022-06-06T01:02:30'
+    assert peak_time.isoformat().split('.')[0] == '2022-06-06T01:02:31'
     assert fig.get_axes()[0].get_title() == 'SOLO/EPT 0.0608 - 0.0678 MeV protons\n5min averaging, viewing: SUN'
     # viewing "north", combined channel, electrons
-    Event1 = Event(spacecraft='Solar Orbiter', sensor='EPT', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
-    # print(Event1.print_energies())  # TODO: see above
+    Event1 = Event(spacecraft='Solar Orbiter', sensor='EPT', viewing='sun', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
+    print(Event1.print_energies())
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='north', background_range=background_range, channels=[1, 4], resample_period="5min", yscale='log', cusum_window=30)
     assert isinstance(flux, pd.Series)
     assert flux.shape == (288,)
     assert len(onset_stats) == 6
     assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)  # onset_stats[5] == pd.Timestamp('2021-10-28 15:31:59.492059')
     assert ~onset_found
-    assert peak_time.isoformat().split('.')[0] == '2022-06-06T23:02:30'
+    assert peak_time.isoformat().split('.')[0] == '2022-06-06T23:02:31'
     assert fig.get_axes()[0].get_title() == 'SOLO/EPT 0.0334 - 0.0439 MeV electrons\n5min averaging, viewing: NORTH'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view='sun')
+    assert Event1.fig.get_axes()[0].get_title() == 'SOLO/EPT (sun) electrons, 2022-06-06'
 
-def test_onset_PSP_ISOIS_EPIHI_online():
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('sun', selection=(0, 4, 1), resample='1min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'Solar Orbiter EPT, electrons'
+
+
+def test_onset_spectrum_tsa_PSP_ISOIS_EPIHI_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 29)
     lpath = f"{os.getcwd()}/data/"
@@ -152,8 +192,17 @@ def test_onset_PSP_ISOIS_EPIHI_online():
     assert peak_time.isoformat().split('.')[0] == '2021-10-28T19:56:59'
     assert fig.get_axes()[0].get_title() == 'PSP/ISOIS-EPIHI 8.0 - 19.0 MeV protons\n5min averaging, viewing: B'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view='A')
+    assert Event1.fig.get_axes()[0].get_title() == 'PSP/ISOIS-EPIHI (A) protons, 2021-10-28'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('A', selection=(0, 4, 1), resample='1min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'Parker Solar Probe ISOIS-EPIHI, protons'
+
 
-def test_onset_PSP_ISOIS_EPILO_e_online():
+def test_onset_spectrum_tsa_PSP_ISOIS_EPILO_e_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 29)
     lpath = f"{os.getcwd()}/data/"
@@ -179,8 +228,17 @@ def test_onset_PSP_ISOIS_EPILO_e_online():
     assert peak_time.isoformat().split('.')[0] == '2021-10-28T17:48:14'
     assert fig.get_axes()[0].get_title() == 'PSP/ISOIS-EPILO 10.0 - 100.5 keV electrons\n5min averaging, viewing: 3'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view='7')
+    assert Event1.fig.get_axes()[0].get_title() == 'PSP/ISOIS-EPILO (7) electrons, 2021-10-28'
 
-def test_onset_Wind_3DP_p_online():
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('3', selection=(0, 4, 1), resample='1min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'Parker Solar Probe ISOIS-EPILO, electrons'
+
+
+def test_onset_spectrum_tsa_Wind_3DP_p_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 29)
     lpath = f"{os.getcwd()}/data/"
@@ -207,12 +265,21 @@ def test_onset_Wind_3DP_p_online():
     assert fig.get_axes()[0].get_title() == 'WIND/3DP 385.96 - 716.78 keV protons\n5min averaging, viewing: OMNIDIRECTIONAL'
     # no channel combination inlcuded for Wind/3DP, yet
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view='sector 3')
+    assert Event1.fig.get_axes()[0].get_title() == 'WIND/3DP (sector 3) protons, 2021-10-28'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('omnidirectional', selection=(0, 4, 1), resample=None)
+    assert plt.figure(1).get_axes()[0].get_title() == 'Wind 3DP, protons'
 
-def test_onset_Wind_3DP_e_online():
+
+def test_onset_spectrum_tsa_Wind_3DP_e_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 29)
     lpath = f"{os.getcwd()}/data/"
-    Event1 = Event(spacecraft='Wind', sensor='3DP', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
+    Event1 = Event(spacecraft='Wind', sensor='3DP', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)  # TODO: radio_spacecraft=('wind', 'WIND')
     print(Event1.print_energies())
     background_range = (datetime.datetime(2021, 10, 28, 10, 0, 0), datetime.datetime(2021, 10, 28, 12, 0, 0))
     #
@@ -235,8 +302,17 @@ def test_onset_Wind_3DP_e_online():
     assert fig.get_axes()[0].get_title() == 'WIND/3DP 127.06 - 235.96 keV electrons\n5min averaging, viewing: OMNIDIRECTIONAL'
     # no channel combination inlcuded for Wind/3DP, yet
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view='omnidirectional')
+    assert Event1.fig.get_axes()[0].get_title() == 'WIND/3DP (omnidirectional) electrons, 2021-10-28'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot('sector 3', selection=(0, 4, 1), resample=None)
+    assert plt.figure(1).get_axes()[0].get_title() == 'Wind 3DP, electrons'
+
 
-def test_onset_STEREOB_HET_p_online():
+def test_onset_spectrum_tsa_STEREOB_HET_p_online():
     startdate = datetime.date(2006, 12, 13)
     enddate = datetime.date(2006, 12, 14)
     lpath = f"{os.getcwd()}/data/"
@@ -253,12 +329,21 @@ def test_onset_STEREOB_HET_p_online():
     assert peak_time.isoformat().split('.')[0] == '2006-12-13T09:53:04'
     assert fig.get_axes()[0].get_title() == 'STB/HET 26.3 - 40.5 MeV protons\n5min averaging'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view=None)
+    assert Event1.fig.get_axes()[0].get_title() == 'STB/HET protons, 2006-12-13'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=None, resample=None)
+    assert plt.figure(1).get_axes()[0].get_title() == 'STEREO-B HET, protons'
+
 
-def test_onset_STEREOB_HET_e_online():
+def test_onset_spectrum_tsa_STEREOB_HET_e_online():
     startdate = datetime.date(2006, 12, 13)
     enddate = datetime.date(2006, 12, 14)
     lpath = f"{os.getcwd()}/data/"
-    Event1 = Event(spacecraft='STEREO-B', sensor='HET', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
+    Event1 = Event(spacecraft='STEREO-B', sensor='HET', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath, radio_spacecraft=('behind', 'STEREO-B'))
     print(Event1.print_energies())
     background_range = (datetime.datetime(2006, 12, 13, 0, 0, 0), datetime.datetime(2006, 12, 13, 2, 0, 0))
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing=None, background_range=background_range, channels=[1], resample_period="5min", yscale='log', cusum_window=30)
@@ -271,8 +356,17 @@ def test_onset_STEREOB_HET_e_online():
     assert peak_time.isoformat().split('.')[0] == '2006-12-13T04:53:04'
     assert fig.get_axes()[0].get_title() == 'STB/HET 1.4 - 2.8 MeV electrons\n5min averaging'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view=None)
+    assert Event1.fig.get_axes()[0].get_title() == 'Radio & Dynamic Spectrum, STB/HET electrons, 2006-12-13'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=None, resample=None)
+    assert plt.figure(1).get_axes()[0].get_title() == 'STEREO-B HET, electrons'
 
-def test_onset_STEREOA_SEPT_p_online():
+
+def test_onset_spectrum_tsa_STEREOA_SEPT_p_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 28)
     lpath = f"{os.getcwd()}/data/"
@@ -289,12 +383,21 @@ def test_onset_STEREOA_SEPT_p_online():
     assert peak_time.isoformat().split('.')[0] == '2021-10-28T17:18:27'
     assert fig.get_axes()[0].get_title() == 'STA/SEPT 110-174.6 keV protons\n5min averaging, viewing: NORTH'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view=None)
+    assert Event1.fig.get_axes()[0].get_title() == 'STA/SEPT protons, 2021-10-28'
 
-def test_onset_STEREOA_SEPT_e_online():
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=None, resample=None)
+    assert plt.figure(1).get_axes()[0].get_title() == 'STEREO-A SEPT, protons'
+
+
+def test_onset_spectrum_tsa_STEREOA_SEPT_e_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 28)
     lpath = f"{os.getcwd()}/data/"
-    Event1 = Event(spacecraft='STEREO-A', sensor='SEPT', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath)
+    Event1 = Event(spacecraft='STEREO-A', sensor='SEPT', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath, radio_spacecraft=('ahead', 'STEREO-A'))
     print(Event1.print_energies())
     background_range = (datetime.datetime(2021, 10, 28, 10, 0, 0), datetime.datetime(2021, 10, 28, 12, 0, 0))
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='asun', background_range=background_range, channels=[8], resample_period="5min", yscale='log', cusum_window=30)
@@ -307,8 +410,17 @@ def test_onset_STEREOA_SEPT_e_online():
     assert peak_time.isoformat().split('.')[0] == '2021-10-28T18:58:27'
     assert fig.get_axes()[0].get_title() == 'STA/SEPT 125-145 keV electrons\n5min averaging, viewing: ASUN'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view=None)
+    assert Event1.fig.get_axes()[0].get_title() == 'Radio & Dynamic Spectrum, STA/SEPT electrons, 2021-10-28'
 
-def test_onset_SOHO_EPHIN_online():
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=None, resample=None)
+    assert plt.figure(1).get_axes()[0].get_title() == 'STEREO-A SEPT, electrons'
+
+
+def test_onset_spectrum_tsa_SOHO_EPHIN_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 28)
     lpath = f"{os.getcwd()}/data/"
@@ -326,12 +438,28 @@ def test_onset_SOHO_EPHIN_online():
     assert fig.get_axes()[0].get_title() == 'SOHO/EPHIN 0.25 - 0.7 MeV electrons\n5min averaging'
     # no channel combination inlcuded for SOHO/EPHIN electrons, yet
 
+    # test dynamic spectrum:
+    # check = False
+    # try:
+    #     Event1.dynamic_spectrum(view=None)
+    # except Warning:
+    #     check = True
+    # assert check
+    Event1.dynamic_spectrum(view=None)
+    assert Event1.fig.get_axes()[0].get_title() == 'SOHO/EPHIN electrons, 2021-10-28'
+
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=(0, 4, 1), resample='5min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'SOHO EPHIN, electrons'
+
 
-def test_onset_SOHO_ERNE_online():
+def test_onset_spectrum_tsa_SOHO_ERNE_online():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 29)
     lpath = f"{os.getcwd()}/data/"
-    Event1 = Event(spacecraft='SOHO', sensor='ERNE-HED', data_level='l2', species='protons', start_date=startdate, end_date=enddate, data_path=lpath)
+    Event1 = Event(spacecraft='SOHO', sensor='ERNE-HED', data_level='l2', species='protons', start_date=startdate, end_date=enddate, data_path=lpath, radio_spacecraft=('ahead', 'STEREO-A'))
     print(Event1.print_energies())
     background_range = (datetime.datetime(2021, 10, 28, 10, 0, 0), datetime.datetime(2021, 10, 28, 12, 0, 0))
     flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing=None, background_range=background_range, channels=3, resample_period="5min", yscale='log', cusum_window=30)
@@ -344,8 +472,18 @@ def test_onset_SOHO_ERNE_online():
     assert peak_time.isoformat().split('.')[0] == '2021-10-28T22:53:05'
     assert fig.get_axes()[0].get_title() == 'SOHO/ERNE 25.0 - 32.0 MeV protons\n5min averaging'
 
+    # test dynamic spectrum:
+    Event1.dynamic_spectrum(view=None)
+
+    assert Event1.fig.get_axes()[0].get_title() == 'Radio & Dynamic Spectrum, SOHO/ERNE protons, 2021-10-28'
+
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=(0, 4, 1), resample='5min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'SOHO ERNE, protons'
 
-def test_onset_SOHO_ERNE_offline():
+
+def test_onset_tsa_SOHO_ERNE_offline():
     startdate = datetime.date(2021, 10, 28)
     enddate = datetime.date(2021, 10, 29)
     fullpath = get_pkg_data_filename('data/test/soho_erne-hed_l2-1min_20211028_v01.cdf', package='seppy')
@@ -363,6 +501,11 @@ def test_onset_SOHO_ERNE_offline():
     assert peak_time.isoformat().split('.')[0] == '2021-10-28T22:53:05'
     assert fig.get_axes()[0].get_title() == 'SOHO/ERNE 16.0 - 32.0 MeV protons\n5min averaging'
 
+    # test tsa plot:
+    plt.close('all')  # in order to pick the right figure, make sure all previous are closed
+    Event1.tsa_plot(None, selection=(0, 4, 1), resample='5min')
+    assert plt.figure(1).get_axes()[0].get_title() == 'SOHO ERNE, protons'
+
 
 def test_dynamic_spectrum_SOHO_ERNE_offline():
     startdate = datetime.date(2021, 10, 28)
@@ -374,3 +517,22 @@ def test_dynamic_spectrum_SOHO_ERNE_offline():
     Event1.dynamic_spectrum(view=None)
 
     assert Event1.fig.get_axes()[0].get_title() == 'SOHO/ERNE protons, 2021-10-28'
+
+
+def test_onset_Bepi_SIXS_offline():
+    startdate = datetime.date(2023, 7, 19)
+    enddate = datetime.date(2023, 7, 19)
+    fullpath = get_pkg_data_filename('data/test/20230719_side1.csv', package='seppy')
+    lpath = Path(fullpath).parent.as_posix()
+    # lpath = '/home/jagies/data/bepi/bc_mpo_sixs/data_csv/cruise/sixs-p/raw'
+    Event1 = Event(spacecraft='Bepi', sensor='SIXS', data_level='l2', species='electrons', start_date=startdate, end_date=enddate, data_path=lpath, viewing='1')
+    background_range = (datetime.datetime(2023, 7, 19, 0, 30, 0), datetime.datetime(2023, 7, 19, 1, 30, 0))
+    flux, onset_stats, onset_found, peak_flux, peak_time, fig, bg_mean = Event1.find_onset(viewing='1', background_range=background_range, channels=2, resample_period="1min", yscale='log', cusum_window=30)
+
+    assert isinstance(flux, pd.Series)
+    assert flux.shape == (161,)
+    assert len(onset_stats) == 6
+    assert isinstance(onset_stats[5], pd._libs.tslibs.nattype.NaTType)
+    assert ~onset_found
+    assert peak_time.isoformat().split('.')[0] == '2023-07-19T02:25:42'
+    assert fig.get_axes()[0].get_title() == 'BEPI/SIXS 106 keV electrons\n1min averaging, viewing: 1'