def test_dpa():
mdl = ThermalModel('dpa', start='2012:001', stop='2012:007',
model_spec=abs_path('dpa.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
mdl.comp['1dpamzt'].set_data(30.0)
mdl.comp['sim_z'].set_data((100000 * sin(times)).astype(int))
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.comp['fep_count'].set_data((abs(sin(times / 7.89)) * 6.5
).astype(int))
mdl.comp['ccd_count'].set_data((abs(sin(times / 3.111)) * 6.5
).astype(int))
mdl.comp['vid_board'].set_data((abs(sin(times / 2.5)) * 1.5
).astype(int))
mdl.comp['clocking'].set_data((abs(cos(times) / 1.4) * 1.5
).astype(int))
mdl.comp['dpa_power'].set_data(60)
mdl.make()
mdl.calc()
dpa = mdl.comp['1dpamzt']
reffile = abs_path('dpa.npz')
if not os.path.exists(reffile):
print 'Writing reference file dpa.npz'
np.savez(reffile, times=mdl.times, dvals=dpa.dvals,
mvals=dpa.mvals)
regr = np.load(reffile)
assert np.allclose(mdl.times, regr['times'])
assert np.allclose(dpa.dvals, regr['dvals'])
assert np.allclose(dpa.mvals, regr['mvals'])
def test_pftank2t():
mdl = ThermalModel('pftank2t', start='2012:001:12:00:00', stop='2012:004:12:00:00',
model_spec=abs_path('pftank2t.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
msids = ('pftank2t', 'pf0tank2t')
for msid in msids:
mdl.comp[msid].set_data(10.0)
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.make()
mdl.calc()
regrfile = abs_path('pftank2t.npz')
if not os.path.exists(regrfile):
print('Writing reference file', regrfile)
kwargs = {msid: mdl.comp[msid].mvals for msid in msids}
np.savez(regrfile, times=mdl.times, **kwargs)
regr = np.load(regrfile)
assert np.allclose(mdl.times, regr['times'])
for msid in msids:
assert np.allclose(mdl.comp[msid].mvals, regr[msid])
# Test that setattr works for component parameter value by changing one
# value and seeing that the model prediction changes substantially.
mvals = mdl.comp['pftank2t'].mvals.copy()
mdl.comp['solarheat__pf0tank2t'].P_60 = 5.0
mdl.calc()
mvals2 = mdl.comp['pftank2t'].mvals
assert np.any(abs(mvals2 - mvals) > 5)
def test_pitch_range_clip():
"""Pitch in this time range goes from approximately 48.5 to 175.0 degrees.
pftank2t.json is used as a placeholder to load a new thermal model, and
should be able to be replaced with any other model. Make sure the pitch
range stored in the model object does not get clipped to a narrower range.
Make sure the pitch range stored does not include values outside of the 45
to 180 degree range.
Parameters
----------
Returns
-------
"""
mdl = ThermalModel('tank', start='2019:120:12:00:00', stop='2019:122:12:00:00',
model_spec=abs_path('pftank2t.json'))
mdl.comp['pf0tank2t'].set_data(20.0)
mdl.make()
mdl.calc()
pitch = mdl.get_comp('pitch')
assert np.any(pitch.mvals > 170)
assert np.all((pitch.mvals > 45) & (pitch.mvals < 180))
def test_data_types():
for data_type in (int, float, np.float32, np.float64, np.int32, np.int64, np.complex64):
mdl = ThermalModel('dpa', start='2012:001:12:00:00', stop='2012:007:12:00:00',
model_spec=abs_path('dpa.json'))
dpa = mdl.comp['1dpamzt']
dpa.set_data(data_type(30.0))
if data_type is np.complex64:
with pytest.raises(ValueError):
dpa.dvals
else:
dpa.dvals # Property should evaluate OK
def get_dpa_model():
mdl = ThermalModel('dpa', start='2012:001', stop='2012:007',
model_spec=abs_path('dpa.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
mdl.comp['1dpamzt'].set_data(30.0)
mdl.comp['sim_z'].set_data((100000 * sin(times)).astype(int))
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.comp['fep_count'].set_data((abs(sin(times / 7.89)) * 6.5
).astype(int))
mdl.comp['ccd_count'].set_data((abs(sin(times / 3.111)) * 6.5
).astype(int))
mdl.comp['vid_board'].set_data((abs(sin(times / 2.5)) * 1.5
).astype(int))
mdl.comp['clocking'].set_data((abs(cos(times) / 1.4) * 1.5
).astype(int))
mdl.comp['dpa_power'].set_data(60)
mdl.make()
mdl.calc()
return mdl
def test_pftank2t():
mdl = ThermalModel('pftank2t', start='2012:001', stop='2012:004',
model_spec=abs_path('pftank2t.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
msids = ('pftank2t', 'pf0tank2t')
for msid in msids:
mdl.comp[msid].set_data(10.0)
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.make()
mdl.calc()
regrfile = abs_path('pftank2t.npz')
if not os.path.exists(regrfile):
print('Writing reference file', regrfile)
kwargs = {msid: mdl.comp[msid].mvals for msid in msids}
np.savez(regrfile, times=mdl.times, **kwargs)
regr = np.load(regrfile)
assert np.allclose(mdl.times, regr['times'])
for msid in msids:
assert np.allclose(mdl.comp[msid].mvals, regr[msid])
def test_minusz():
mdl = ThermalModel('minusz', start='2012:001', stop='2012:004',
model_spec=abs_path('minusz.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
msids = ('tephin', 'tcylaft6', 'tcylfmzm', 'tmzp_my', 'tfssbkt1')
for msid in msids:
mdl.comp[msid].set_data(10.0)
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.make()
mdl.calc()
regrfile = 'minusz.npz'
if not os.path.exists(regrfile):
print 'Writing reference file', regrfile
kwargs = {msid: mdl.comp[msid].mvals for msid in msids}
np.savez(regrfile, times=mdl.times, **kwargs)
regr = np.load(regrfile)
assert np.allclose(mdl.times, regr['times'])
for msid in msids:
assert np.allclose(mdl.comp[msid].mvals, regr[msid])
def test_dpa_remove_pow():
mdl = ThermalModel('dpa', start='2019:001:12:00:00', stop='2019:007:12:00:00',
model_spec=abs_path('dpa_remove_pow.json'))
mdl._get_cmd_states()
mdl.make()
mdl.calc()
dpa = mdl.comp['1dpamzt']
reffile = abs_path('dpa_remove_pow.npz')
if not os.path.exists(reffile):
print('Writing reference file', reffile)
np.savez(reffile, times=mdl.times, dvals=dpa.dvals,
mvals=dpa.mvals)
regr = np.load(reffile)
assert np.allclose(mdl.times, regr['times'])
assert np.allclose(dpa.dvals, regr['dvals'])
assert np.allclose(dpa.mvals, regr['mvals'])
def test_pitch_clip():
"""Pitch in this time range goes from 48.62 to 158.41. dpa_clip.json
has been modified so the solarheat pitch range is from 55 .. 153.
Make sure the model still runs with no interpolation error.
Parameters
----------
Returns
-------
"""
mdl = ThermalModel('dpa', start='2012:001:12:00:00', stop='2012:007:12:00:00',
model_spec=abs_path('dpa_clip.json'))
mdl._get_cmd_states()
mdl.make()
mdl.calc()
def test_pitch_clip():
"""
Pitch in this time range goes from 48.62 to 158.41. dpa_clip.json
has been modified so the solarheat pitch range is from 55 .. 153.
Make sure the model still runs with no interpolation error.
"""
mdl = ThermalModel('dpa', start='2012:001', stop='2012:007',
model_spec=abs_path('dpa_clip.json'))
try:
mdl._get_cmd_states()
except:
pytest.skip('No commanded states access - '
'cannot run DPA model with real states')
mdl.make()
mdl.calc()
def test_dpa_remove_pow():
mdl = ThermalModel('dpa', start='2012:001', stop='2012:007',
model_spec=abs_path('dpa_remove_pow.json'))
# Check that cmd_states database can be read. Skip if not, probably
# running test on a platform without access.
try:
mdl._get_cmd_states()
except:
pytest.skip('No commanded states access - '
'cannot run DPA model with real states')
mdl.make()
mdl.calc()
dpa = mdl.comp['1dpamzt']
reffile = abs_path('dpa_remove_pow.npz')
if not os.path.exists(reffile):
print('Writing reference file', reffile)
np.savez(reffile, times=mdl.times, dvals=dpa.dvals,
mvals=dpa.mvals)
regr = np.load(reffile)
assert np.allclose(mdl.times, regr['times'])
assert np.allclose(dpa.dvals, regr['dvals'])
assert np.allclose(dpa.mvals, regr['mvals'])
def get_dpa_model():
mdl = ThermalModel('dpa', start='2012:001:12:00:00', stop='2012:007:12:00:00',
model_spec=abs_path('dpa.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
mdl.comp['1dpamzt'].set_data(30.0)
mdl.comp['sim_z'].set_data((100000 * sin(times)).astype(int))
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.comp['fep_count'].set_data((abs(sin(times / 7.89)) * 6.5
).astype(int))
mdl.comp['ccd_count'].set_data((abs(sin(times / 3.111)) * 6.5
).astype(int))
mdl.comp['vid_board'].set_data((abs(sin(times / 2.5)) * 1.5
).astype(int))
mdl.comp['clocking'].set_data((abs(cos(times) / 1.4) * 1.5
).astype(int))
mdl.comp['dpa_power'].set_data(60)
mdl.make()
mdl.calc()
return mdl
def test_minusz():
mdl = ThermalModel('minusz', start='2012:001:12:00:00', stop='2012:004:12:00:00',
model_spec=abs_path('minusz.json'))
times = (mdl.times - mdl.times[0]) / 10000.0
msids = ('tephin', 'tcylaft6', 'tcylfmzm', 'tmzp_my', 'tfssbkt1')
for msid in msids:
mdl.comp[msid].set_data(10.0)
mdl.comp['pitch'].set_data(45.1 + 55 * (1 + sin(times / 2)))
mdl.comp['eclipse'].set_data(cos(times) > 0.95)
mdl.make()
mdl.calc()
regrfile = abs_path('minusz.npz')
if not os.path.exists(regrfile):
print('Writing reference file', regrfile)
kwargs = {msid: mdl.comp[msid].mvals for msid in msids}
np.savez(regrfile, times=mdl.times, **kwargs)
regr = np.load(regrfile)
assert np.allclose(mdl.times, regr['times'])
for msid in msids:
assert np.allclose(mdl.comp[msid].mvals, regr[msid])
def test_multi_solar_heat_values():
P_pitches = [45, 180]
P_vals = [1.0, 1.0]
P_pitches2 = [45, 65, 90, 140, 180]
P_vals2 = [1.0, 1.0, 0.0, 1.0, 1.0]
model = ThermalModel('test', start='2011:001', stop='2011:005')
tephin = model.add(Node, 'tephin')
tcylaft6 = model.add(Node, 'tcylaft6')
pitch = model.add(Pitch)
eclipse = model.add(Eclipse)
model.add(HeatSink, tephin, T=0.0, tau=200.0)
model.add(HeatSink, tcylaft6, T=0.0, tau=200.0)
model.add(SolarHeat, tephin, pitch, eclipse, P_pitches, P_vals)
model.add(SolarHeat, tcylaft6, pitch, eclipse, P_pitches2, P_vals2)
tephin.set_data(30.0)
tcylaft6.set_data(30.0)
pitch.set_data(90.0)
eclipse.set_data(False)
model.make()
model.calc()
mvals = model.comp['tephin'].mvals
mvals2 = model.comp['tcylaft6'].mvals
assert len(mvals) == 1051
assert mvals[0] == 30.0
assert abs(mvals[500] - 157.4740) < 0.001
assert abs(mvals[1050] - 196.4138) < 0.001
assert len(mvals2) == 1051
assert mvals2[0] == 30.0
assert abs(mvals2[500] - 15.8338) < 0.001
assert abs(mvals2[1050] - 11.4947) < 0.001
# Make sure we can round-trip the model through a JSON file
temp_name = tempfile.NamedTemporaryFile(delete=False).name
model.write(temp_name)
model2 = ThermalModel('test', model_spec=temp_name,
start='2011:001', stop='2011:005')
os.unlink(temp_name)
model2.get_comp('tephin').set_data(30.0)
model2.get_comp('tcylaft6').set_data(30.0)
model2.get_comp('pitch').set_data(90.0)
model2.get_comp('eclipse').set_data(False)
model2.make()
model2.calc()
mvals = model2.comp['tephin'].mvals
mvals2 = model2.comp['tcylaft6'].mvals
assert len(mvals) == 1051
assert mvals[0] == 30.0
assert abs(mvals[500] - 157.4740) < 0.001
assert abs(mvals[1049] - 196.4138) < 0.001
assert len(mvals2) == 1051
assert mvals2[0] == 30.0
assert abs(mvals2[500] - 15.8338) < 0.001
assert abs(mvals2[1049] - 11.4947) < 0.001
def test_multi_solar_heat_values():
P_pitches = [45, 180]
P_vals = [1.0, 1.0]
P_pitches2 = [45, 65, 90, 140, 180]
P_vals2 = [1.0, 1.0, 0.0, 1.0, 1.0]
model = ThermalModel('test', start='2011:001:12:00:00', stop='2011:005:12:00:00')
tephin = model.add(Node, 'tephin')
tcylaft6 = model.add(Node, 'tcylaft6')
pitch = model.add(Pitch)
eclipse = model.add(Eclipse)
model.add(HeatSink, tephin, T=0.0, tau=200.0)
model.add(HeatSink, tcylaft6, T=0.0, tau=200.0)
model.add(SolarHeat, tephin, pitch, eclipse, P_pitches, P_vals)
model.add(SolarHeat, tcylaft6, pitch, eclipse, P_pitches2, P_vals2)
tephin.set_data(30.0)
tcylaft6.set_data(30.0)
pitch.set_data(90.0)
eclipse.set_data(False)
model.make()
model.calc()
mvals = model.comp['tephin'].mvals
mvals2 = model.comp['tcylaft6'].mvals
assert len(mvals) == 1051
assert mvals[0] == 30.0
assert abs(mvals[500] - 157.4740) < 0.001
assert abs(mvals[1050] - 196.4138) < 0.001
assert len(mvals2) == 1051
assert mvals2[0] == 30.0
assert abs(mvals2[500] - 15.8338) < 0.001
assert abs(mvals2[1050] - 11.4947) < 0.001
# Make sure we can round-trip the model through a JSON file
temp_name = tempfile.NamedTemporaryFile(delete=False).name
model.write(temp_name)
model2 = ThermalModel('test', model_spec=temp_name,
start='2011:001:12:00:00', stop='2011:005:12:00:00')
os.unlink(temp_name)
model2.get_comp('tephin').set_data(30.0)
model2.get_comp('tcylaft6').set_data(30.0)
model2.get_comp('pitch').set_data(90.0)
model2.get_comp('eclipse').set_data(False)
model2.make()
model2.calc()
mvals = model2.comp['tephin'].mvals
mvals2 = model2.comp['tcylaft6'].mvals
assert len(mvals) == 1051
assert mvals[0] == 30.0
assert abs(mvals[500] - 157.4740) < 0.001
assert abs(mvals[1049] - 196.4138) < 0.001
assert len(mvals2) == 1051
assert mvals2[0] == 30.0
assert abs(mvals2[500] - 15.8338) < 0.001
assert abs(mvals2[1049] - 11.4947) < 0.001
