def test_load_hro_data(self):
omni_vars = pyspedas.omni.data(level='hro')
self.assertTrue(data_exists('BX_GSE'))
self.assertTrue(data_exists('BY_GSE'))
self.assertTrue(data_exists('BZ_GSE'))
self.assertTrue(data_exists('BY_GSM'))
self.assertTrue(data_exists('BZ_GSM'))
self.assertTrue(data_exists('proton_density'))
def test_load_gmag_data(self):
data = pyspedas.themis.gmag()
self.assertTrue(data_exists('thg_mag_amer'))
self.assertTrue(data_exists('thg_mag_kil'))
self.assertTrue(data_exists('thg_mag_eagl'))
self.assertTrue(data_exists('thg_mag_leth'))
self.assertTrue(data_exists('thg_mag_inuv'))
self.assertTrue(data_exists('thg_mag_wrth'))
def test_default(self):
data = pyspedas.mms.fgm(
probe=[1, 2, 3, 4],
trange=['2015-10-30/05:15:45', '2015-10-30/05:15:48'],
data_rate='brst',
time_clip=True,
varformat='*_gse_*',
get_fgm_ephemeris=True)
curl = pyspedas.mms.curlometer(positions=[
'mms1_fgm_r_gse_brst_l2', 'mms2_fgm_r_gse_brst_l2',
'mms3_fgm_r_gse_brst_l2', 'mms4_fgm_r_gse_brst_l2'
],
fields=[
'mms1_fgm_b_gse_brst_l2',
'mms2_fgm_b_gse_brst_l2',
'mms3_fgm_b_gse_brst_l2',
'mms4_fgm_b_gse_brst_l2'
])
self.assertTrue(data_exists('jtotal'))
self.assertTrue(data_exists('curlB'))
self.assertTrue(data_exists('divB'))
self.assertTrue(data_exists('jpar'))
self.assertTrue(data_exists('jperp'))
def test_pad_extof_srvy(self):
mms_load_eis(datatype='extof')
mms_eis_pad(datatype='extof')
self.assertTrue(data_exists('mms1_epd_eis_srvy_l2_extof_46-10489keV_proton_flux_omni_pad_spin'))
self.assertTrue(data_exists('mms1_epd_eis_srvy_l2_extof_46-10489keV_proton_flux_omni_pad'))
self.assertTrue(data_exists('mms1_epd_eis_srvy_l2_extof_proton_flux_omni_spin'))
self.assertTrue(data_exists('mms1_epd_eis_srvy_l2_extof_proton_flux_omni'))
self.assertTrue(data_exists('mms1_epd_eis_srvy_l2_extof_oxygen_energy_range'))
self.assertTrue(data_exists('mms1_epd_eis_srvy_l2_extof_proton_energy_range'))
def test_pad_extof_srvy_probe(self):
mms_load_eis(probe=4)
mms_eis_pad(probe=4)
self.assertTrue(data_exists('mms4_epd_eis_srvy_l2_extof_44-1315keV_proton_flux_omni_pad_spin'))
self.assertTrue(data_exists('mms4_epd_eis_srvy_l2_extof_44-1315keV_proton_flux_omni_pad'))
self.assertTrue(data_exists('mms4_epd_eis_srvy_l2_extof_proton_flux_omni_spin'))
self.assertTrue(data_exists('mms4_epd_eis_srvy_l2_extof_proton_flux_omni'))
self.assertTrue(data_exists('mms4_epd_eis_srvy_l2_extof_oxygen_energy_range'))
self.assertTrue(data_exists('mms4_epd_eis_srvy_l2_extof_proton_energy_range'))
def mms_split_fgm_data(probe, data_rate, level, instrument, suffix=''):
"""
"""
probe = probe.lower()
instrument = instrument.lower()
data_rate = data_rate.lower()
level = level.lower()
if level.lower() == 'l2pre':
data_rate_mod = data_rate + '_l2pre'
else:
data_rate_mod = data_rate
coords = ['dmpa', 'gse', 'gsm', 'bcs']
out_vars = []
for coord in coords:
if level in ['l2', 'l2pre']:
tplot_name = 'mms' + probe + '_' + instrument + '_b_' + coord + '_' + data_rate + '_' + level + suffix
else:
tplot_name = 'mms' + probe + '_' + instrument + '_' + data_rate_mod + '_' + coord + suffix
if data_exists(tplot_name) == False:
continue
fgm_data = get_data(tplot_name)
if fgm_data is None:
continue
metadata = get_data(tplot_name, metadata=True)
if suffix != '':
tplot_name = tplot_name[0:-len(suffix)]
store_data(tplot_name + '_bvec' + suffix, data={'x': fgm_data.times, 'y': fgm_data.y[:, :3]}, attr_dict=metadata)
store_data(tplot_name + '_btot' + suffix, data={'x': fgm_data.times, 'y': fgm_data.y[:, 3]}, attr_dict=metadata)
options(tplot_name + '_btot' + suffix, 'legend_names', 'Bmag')
options(tplot_name + '_btot' + suffix, 'ytitle', 'MMS'+probe + ' FGM')
out_vars.append(tplot_name + '_bvec' + suffix)
out_vars.append(tplot_name + '_btot' + suffix)
return out_vars
def test_load_brst_multi_probe(self):
del_data('*')
data = mms_load_feeps(trange=['2015-10-16/13:06', '2015-10-16/13:07'],
data_rate='brst',
probe=[1, 2, 3, 4])
self.assertTrue(
data_exists('mms1_epd_feeps_brst_l2_electron_intensity_omni'))
self.assertTrue(
data_exists('mms1_epd_feeps_brst_l2_electron_intensity_omni_spin'))
self.assertTrue(
data_exists('mms2_epd_feeps_brst_l2_electron_intensity_omni'))
self.assertTrue(
data_exists('mms2_epd_feeps_brst_l2_electron_intensity_omni_spin'))
self.assertTrue(
data_exists('mms3_epd_feeps_brst_l2_electron_intensity_omni'))
self.assertTrue(
data_exists('mms3_epd_feeps_brst_l2_electron_intensity_omni_spin'))
self.assertTrue(
data_exists('mms4_epd_feeps_brst_l2_electron_intensity_omni'))
self.assertTrue(
data_exists('mms4_epd_feeps_brst_l2_electron_intensity_omni_spin'))
def test_load_emfisis_data(self):
emfisis_vars = pyspedas.rbsp.emfisis(trange=['2018-11-5', '2018-11-6'],
datatype='magnetometer',
level='l3',
time_clip=True)
wfr_vars = pyspedas.rbsp.emfisis(trange=['2018-11-5', '2018-11-6'],
level='l2',
datatype='wfr')
hfr_vars = pyspedas.rbsp.emfisis(trange=['2018-11-5', '2018-11-6'],
level='l2',
datatype='hfr')
self.assertTrue(data_exists('Mag'))
# WFR waveform data
self.assertTrue(data_exists('BuSamples'))
self.assertTrue(data_exists('BvSamples'))
self.assertTrue(data_exists('BwSamples'))
self.assertTrue(data_exists('EuSamples'))
self.assertTrue(data_exists('EvSamples'))
self.assertTrue(data_exists('EwSamples'))
# HFR waveform data
self.assertTrue(data_exists('HFRsamples'))
def test_load_lena_data(self):
lena_vars = pyspedas.image.lena(time_clip=True)
self.assertTrue(data_exists('Image0'))
def test_sroi(self):
sroi = mms_load_sroi_segments(trange=['2019-10-01', '2019-11-01'])
self.assertTrue(data_exists('mms1_bss_sroi'))
self.assertTrue(len(sroi[0]) == 28)
self.assertTrue(sroi[0][0] == 1569849345.0)
self.assertTrue(sroi[1][0] == 1569923029.0)
def test_load_mgf_data(self):
mgf_vars = pyspedas.erg.mgf(time_clip=True)
self.assertTrue(data_exists('erg_mgf_l2_mag_8sec_sm'))
def test_load_efw_data(self):
efw_vars = pyspedas.cluster.efw()
self.assertTrue(data_exists('E_pow_f1__C1_PP_EFW'))
def test_fast(self):
fast = mms_load_fast_segments(trange=['2015-10-01', '2015-11-01'])
self.assertTrue(data_exists('mms_bss_fast'))
self.assertTrue(len(fast[0]) == 35)
self.assertTrue(fast[0][0] == 1443504404.0)
self.assertTrue(fast[1][0] == 1443554774.0)
def test_load_euv_data(self):
euv_vars = pyspedas.image.euv()
self.assertTrue(data_exists('IMAGE'))
def test_load_preorb_data(self):
orb_vars = pyspedas.image.orbit(datatype='pre_or')
self.assertTrue(data_exists('GSM_POS'))
self.assertTrue(data_exists('GSM_VEL'))
def test_load_cis_data(self):
cis_vars = pyspedas.cluster.cis()
self.assertTrue(data_exists('N_p__C1_PP_CIS'))
def test_load_efw_data(self):
efw_vars = pyspedas.rbsp.efw(trange=['2015-11-5', '2015-11-6'],
level='l3')
self.assertTrue(data_exists('density'))
def test_load_whi_data(self):
whi_vars = pyspedas.cluster.whi()
self.assertTrue(data_exists('E_pow_f5__C1_PP_WHI'))
def test_load_fgm_data(self):
mag_vars = pyspedas.cluster.fgm()
self.assertTrue(data_exists('B_xyz_gse__C1_UP_FGM'))
def test_load_sta_data(self):
sta_vars = pyspedas.cluster.staff()
self.assertTrue(data_exists('E_pow_f2__C1_PP_STA'))
def test_load_rap_data(self):
r_vars = pyspedas.cluster.rapid()
self.assertTrue(data_exists('J_e_lo__C1_PP_RAP'))
def test_load_peace_data(self):
p_vars = pyspedas.cluster.peace()
self.assertTrue(data_exists('T_e_par__C1_PP_PEA'))
def test_load_hena_data(self):
hena_vars = pyspedas.image.hena()
self.assertTrue(data_exists('Image0'))
def test_load_mageis_data(self):
mageis_vars = pyspedas.rbsp.mageis(trange=['2018-11-5', '2018-11-6'],
level='l3',
rel='rel04')
self.assertTrue(data_exists('I'))
def test_load_rpi_data(self):
rpi_vars = pyspedas.image.rpi()
self.assertTrue(data_exists('Amplitude'))
def test_load_hope_data(self):
hope_vars = pyspedas.rbsp.hope(trange=['2018-11-5', '2018-11-6'],
datatype='moments',
level='l3',
rel='rel04')
self.assertTrue(data_exists('Ion_density'))
def test_load_fuv_data(self):
fuv_vars = pyspedas.image.fuv()
self.assertTrue(data_exists('WIC_PIXELS'))
def test_brst(self):
brst = mms_load_brst_segments(trange=['2015-10-16', '2015-10-17'])
self.assertTrue(len(brst[0]) == 53)
self.assertTrue(brst[0][0] == 1444975174.0)
self.assertTrue(brst[1][0] == 1444975244.0)
self.assertTrue(data_exists('mms_bss_burst'))
def test_load_dst_data(self):
dst_vars = pyspedas.kyoto.dst()
self.assertTrue(data_exists('kyoto_dst'))
def test_load_dwp_data(self):
dwp_vars = pyspedas.cluster.dwp()
self.assertTrue(data_exists('Correl_freq__C1_PP_DWP'))
