def test_temp_em():
# Create GOESLightcurve object, then create new one with
# temperature & EM using with temp_em().
goeslc = lc.GOESLightCurve.create("2014-01-01 00:00", "2014-01-01 01:00")
goeslc_new = goes.temp_em(goeslc)
# Test correct exception is raised if a GOESLightCurve object is
# not inputted.
with pytest.raises(TypeError):
goes.temp_em([])
# Find temperature and EM manually with goes_chianti_tem()
t, em = goes.goes_chianti_tem(np.array(goeslc.data.xrsb),
np.array(goeslc.data.xrsa),
satellite = \
int(goeslc.meta["TELESCOP"].split()[1]),
date="2014-01-01")
# Check that temperature and EM arrays from goes_chianti_tem()
# are same as those in new GOESLightcurve object.
assert goeslc_new.data.temperature.all() == t.all()
assert goeslc_new.data.em.all() == em.all()
# Check rest of data frame of new GOESLightCurve object is same
# as that in original object.
goeslc_revert = copy.deepcopy(goeslc_new)
del goeslc_revert.data["temperature"]
del goeslc_revert.data["em"]
assert_frame_equal(goeslc_revert.data, goeslc.data)
def test_temp_em():
# Create GOESLightcurve object, then create new one with
# temperature & EM using with temp_em().
goeslc = lc.GOESLightCurve.create("2014-01-01 00:00", "2014-01-01 01:00")
goeslc_new = goes.temp_em(goeslc)
# Test correct exception is raised if a GOESLightCurve object is
# not inputted.
with pytest.raises(TypeError):
goes.temp_em([])
# Find temperature and EM manually with goes_chianti_tem()
t, em = goes.goes_chianti_tem(np.array(goeslc.data.xrsb),
np.array(goeslc.data.xrsa),
satellite = \
int(goeslc.meta["TELESCOP"].split()[1]),
date="2014-01-01")
# Check that temperature and EM arrays from goes_chianti_tem()
# are same as those in new GOESLightcurve object.
assert goeslc_new.data.temperature.all() == t.all()
assert goeslc_new.data.em.all() == em.all()
# Check rest of data frame of new GOESLightCurve object is same
# as that in original object.
goeslc_revert = copy.deepcopy(goeslc_new)
del goeslc_revert.data["temperature"]
del goeslc_revert.data["em"]
assert_frame_equal(goeslc_revert.data, goeslc.data)
def test_goes_chianti_tem():
longflux = np.array([7e-6])
shortflux = np.array([7e-7])
ratio = shortflux/longflux
shortflux_toomany = np.append(shortflux, shortflux[0])
shortflux_toosmall = copy.deepcopy(shortflux)
shortflux_toosmall[0] = -1
shortflux_toobig = copy.deepcopy(shortflux)
shortflux_toobig[0] = 1
temp_test = np.zeros(len(longflux))+10
temp_test_toomany = np.append(temp_test, 0)
temp_test_toosmall = copy.deepcopy(temp_test)
temp_test_toosmall[0] = -1
temp_test_toobig = copy.deepcopy(temp_test)
temp_test_toobig[0] = 101
date = "2014-04-16"
# First test correct exceptions are raised if incorrect inputs are
# entered.
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux, satellite=-1)
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux_toomany)
with pytest.raises(ValueError):
temp = goes._goes_get_chianti_temp(ratio, satellite=-1)
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux,
abundances="Neither")
with pytest.raises(ValueError):
temp = goes._goes_get_chianti_temp(ratio, abundances="Neither")
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux_toobig)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test, satellite=-1)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test,
abundances="Neither")
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test_toomany)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test_toosmall)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test_toobig)
# test case 1: satellite > 7, abundances = coronal
temp1, em1 = goes.goes_chianti_tem(longflux, shortflux, satellite=15,
date=date)
assert np.around(temp1[0], decimals=2) == 11.28
assert em1[0] < 4.79e+48 and em1[0] > 4.78e+48
# test case 2: satellite > 7, abundances = photospheric
temp2, em2 = goes.goes_chianti_tem(longflux, shortflux, satellite=15,
date=date,
abundances="photospheric")
assert temp2[0] < 10.25 and temp2[0] > 10.24
assert em2[0] < 1.12e+49 and em2[0] > 1.11e+49
# test case 3: satellite < 8 and != 6, abundances = coronal
temp3, em3 = goes.goes_chianti_tem(longflux, shortflux, satellite=5,
date=date,
abundances="coronal")
assert temp3[0] < 11.43 and temp3[0] > 11.42
assert em3[0] < 3.85e+48 and em3[0] > 3.84e+48
# test case 4: satellite < 8 and != 6, abundances = photospheric
temp4, em4 = goes.goes_chianti_tem(longflux, shortflux, satellite=5,
date=date,
abundances="photospheric")
assert temp4[0] < 10.42 and temp4[0] > 10.41
assert em4[0] < 8.81e+48 and em4[0] > 8.80e+48
# test case 5: satellite = 6, date < 1983-06-28, abundances = coronal
temp5, em5 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date="1983-06-27",
abundances="coronal")
assert temp5[0] < 12.30 and temp5[0] > 12.29
assert em5[0] < 3.13e+48 and em5[0] > 3.12e+48
# test case 6: satellite = 6, date < 1983-06-28, abundances = photospheric
temp6, em6 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date="1983-06-27",
abundances="photospheric")
assert temp6[0] < 11.44 and temp6[0] > 11.43
assert em6[0] < 6.74e+48 and em6[0] > 6.73e+48
# test case 7: satellite = 6, date > 1983-06-28, abundances = coronal
temp7, em7 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date=date,
abundances="coronal")
assert temp7[0] < 11.34 and temp7[0] > 11.33
assert em7[0] < 4.08e+48 and em7[0] > 4.07e+48
# test case 8: satellite = 6, date > 1983-06-28, abundances = photospheric
temp8, em8 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date=date,
abundances="photospheric")
assert temp8[0] < 10.36 and temp8[0] > 10.35
assert em8[0] < 9.39e+48 and em8[0] > 9.38e+48
def test_goes_chianti_tem():
longflux = np.array([7e-6])
shortflux = np.array([7e-7])
ratio = shortflux/longflux
shortflux_toomany = np.append(shortflux, shortflux[0])
shortflux_toosmall = copy.deepcopy(shortflux)
shortflux_toosmall[0] = -1
shortflux_toobig = copy.deepcopy(shortflux)
shortflux_toobig[0] = 1
temp_test = np.zeros(len(longflux))+10
temp_test_toomany = np.append(temp_test, 0)
temp_test_toosmall = copy.deepcopy(temp_test)
temp_test_toosmall[0] = -1
temp_test_toobig = copy.deepcopy(temp_test)
temp_test_toobig[0] = 101
date = "2014-04-16"
# First test correct exceptions are raised if incorrect inputs are
# entered.
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux, satellite=-1)
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux_toomany)
with pytest.raises(ValueError):
temp = goes._goes_get_chianti_temp(ratio, satellite=-1)
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux,
abundances="Neither")
with pytest.raises(ValueError):
temp = goes._goes_get_chianti_temp(ratio, abundances="Neither")
with pytest.raises(ValueError):
temp, em = goes.goes_chianti_tem(longflux, shortflux_toobig)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test, satellite=-1)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test,
abundances="Neither")
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test_toomany)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test_toosmall)
with pytest.raises(ValueError):
em = goes._goes_get_chianti_em(longflux, temp_test_toobig)
# test case 1: satellite > 7, abundances = coronal
temp1, em1 = goes.goes_chianti_tem(longflux, shortflux, satellite=15,
date=date)
assert np.around(temp1[0], decimals=2) == 11.28
assert em1[0] < 4.79e+48 and em1[0] > 4.78e+48
# test case 2: satellite > 7, abundances = photospheric
temp2, em2 = goes.goes_chianti_tem(longflux, shortflux, satellite=15,
date=date,
abundances="photospheric")
assert temp2[0] < 10.25 and temp2[0] > 10.24
assert em2[0] < 1.12e+49 and em2[0] > 1.11e+49
# test case 3: satellite < 8 and != 6, abundances = coronal
temp3, em3 = goes.goes_chianti_tem(longflux, shortflux, satellite=5,
date=date,
abundances="coronal")
assert temp3[0] < 11.43 and temp3[0] > 11.42
assert em3[0] < 3.85e+48 and em3[0] > 3.84e+48
# test case 4: satellite < 8 and != 6, abundances = photospheric
temp4, em4 = goes.goes_chianti_tem(longflux, shortflux, satellite=5,
date=date,
abundances="photospheric")
assert temp4[0] < 10.42 and temp4[0] > 10.41
assert em4[0] < 8.81e+48 and em4[0] > 8.80e+48
# test case 5: satellite = 6, date < 1983-06-28, abundances = coronal
temp5, em5 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date="1983-06-27",
abundances="coronal")
assert temp5[0] < 12.30 and temp5[0] > 12.29
assert em5[0] < 3.13e+48 and em5[0] > 3.12e+48
# test case 6: satellite = 6, date < 1983-06-28, abundances = photospheric
temp6, em6 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date="1983-06-27",
abundances="photospheric")
assert temp6[0] < 11.44 and temp6[0] > 11.43
assert em6[0] < 6.74e+48 and em6[0] > 6.73e+48
# test case 7: satellite = 6, date > 1983-06-28, abundances = coronal
temp7, em7 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date=date,
abundances="coronal")
assert temp7[0] < 11.34 and temp7[0] > 11.33
assert em7[0] < 4.08e+48 and em7[0] > 4.07e+48
# test case 8: satellite = 6, date > 1983-06-28, abundances = photospheric
temp8, em8 = goes.goes_chianti_tem(longflux, shortflux, satellite=6,
date=date,
abundances="photospheric")
assert temp8[0] < 10.36 and temp8[0] > 10.35
assert em8[0] < 9.39e+48 and em8[0] > 9.38e+48
