def test_convert_underscores_to_hyphens(self):
params = {}
segment1 = {'x': [], 'y': [], 'yerr': [], 'norm_constant': 1.0}
segment2 = {'x': [], 'y': [], 'yerr': [], 'norm_constant': 1.0}
segment3 = {'x': [], 'y': [], 'yerr': [], 'norm_constant': 1.0}
params['segments'] = [segment1, segment2, segment3]
params['norm_operator'] = '0'
params['y0'] = '1.0'
params['xmin'] = 'min'
params['xmax'] = 'max'
params['stats'] = 'avg'
params['integrate'] = 'true'
payload = DictionaryClass(params)
for keys in payload.get_dict().keys():
if set(keys) & set("_"):
self.fail("Found a '_' in '%s' key." %keys)
segments = payload.get_dict()['segments']
for seg in segments:
for key in seg.keys():
if set(key) & set("_"):
self.fail("Found a '_' in '%s' key." %key)
self.assertEqual(len(payload.__dict__.keys()), 7)
self.assertEquals(len(segments), 3)
def test_dictionary_class(self):
obj = DictionaryClass({'foo' : 'bar', 'nested' : {'myvar' : 5}})
self.assertEqual('bar', obj.foo)
self.assertEqual(5, obj.nested.myvar)
self.assertEqual({'foo' : 'bar', 'nested' : {'myvar' : 5}}, obj.get_dict())
def stack_normalize(private_key, sender_id, msg_id, mtype, params,
extra):
try:
info("stack_normalize()")
try:
payload = DictionaryClass(params)
seds = []
for segment in payload.segments:
x = decode_string(segment.x)
y = decode_string(segment.y)
yerr = decode_string(segment.yerr)
id_ = str(segment.id)
seds.append(IrisSed(x=x, y=y, yerr=yerr, id=id_))
stack = IrisStack(seds)
result = normalize(stack, payload)
for i, segment in enumerate(payload.segments):
segment.x = encode_string(result[i].x)
segment.y = encode_string(result[i].y)
segment.yerr = encode_string(result[i].yerr)
segment.norm_constant = str(result[i].norm_constant)
payload.excludeds = result.excluded
reply_success(msg_id, mtype, payload.get_dict())
except Exception, e:
reply_error(msg_id, sedexceptions.SEDException, e, mtype)
return
except Exception:
error(str(capture_exception()))
def stack_redshift(private_key, sender_id, msg_id, mtype, params,
extra):
try:
info("stack_redshift()")
try:
payload = DictionaryClass(params)
seds = []
for segment in payload.segments:
x = decode_string(segment.x)
y = decode_string(segment.y)
yerr = decode_string(segment.yerr)
z = float(segment.z)
id_ = str(segment.id)
seds.append(IrisSed(x=x, y=y, yerr=yerr, z=z, id=id_))
z0 = float(payload.z0)
correct_flux = payload.correct_flux == "true"
result = redshift(IrisStack(seds), z0, correct_flux)
for i, segment in enumerate(payload.segments):
segment.x = encode_string(result[i].x)
segment.y = encode_string(result[i].y)
segment.yerr = encode_string(result[i].yerr)
payload.excludeds = result.excluded
reply_success(msg_id, mtype, payload.get_dict())
except Exception, e:
reply_error(msg_id, sedexceptions.SEDException, e, mtype)
return
except Exception:
error(str(capture_exception()))
def stack_normalize(private_key, sender_id, msg_id, mtype, params, extra):
try:
info("stack_normalize()")
try:
payload = DictionaryClass(params)
seds = []
for segment in payload.segments:
x = decode_string(segment.x)
y = decode_string(segment.y)
yerr = decode_string(segment.yerr)
id_ = str(segment.id)
seds.append(IrisSed(x=x, y=y, yerr=yerr, id=id_))
stack = IrisStack(seds)
result = normalize(stack, payload)
for i, segment in enumerate(payload.segments):
segment.x = encode_string(result[i].x)
segment.y = encode_string(result[i].y)
segment.yerr = encode_string(result[i].yerr)
segment.norm_constant = str(result[i].norm_constant)
payload.excludeds = result.excluded
reply_success(msg_id, mtype, payload.get_dict())
except Exception, e:
reply_error(msg_id, sedexceptions.SEDException, e, mtype)
return
except Exception:
error(str(capture_exception()))
def stack_redshift(private_key, sender_id, msg_id, mtype, params, extra):
try:
info("stack_redshift()")
try:
payload = DictionaryClass(params)
seds = []
for segment in payload.segments:
x = decode_string(segment.x)
y = decode_string(segment.y)
yerr = decode_string(segment.yerr)
z = float(segment.z)
id_ = str(segment.id)
seds.append(IrisSed(x=x, y=y, yerr=yerr, z=z, id=id_))
z0 = float(payload.z0)
correct_flux = payload.correct_flux == "true"
result = redshift(IrisStack(seds), z0, correct_flux)
for i, segment in enumerate(payload.segments):
segment.x = encode_string(result[i].x)
segment.y = encode_string(result[i].y)
segment.yerr = encode_string(result[i].yerr)
payload.excludeds = result.excluded
reply_success(msg_id, mtype, payload.get_dict())
except Exception, e:
reply_error(msg_id, sedexceptions.SEDException, e, mtype)
return
except Exception:
error(str(capture_exception()))
def stack_stack(private_key, sender_id, msg_id, mtype, params, extra):
try:
info("stack_stack()")
try:
payload = DictionaryClass(params)
seds = []
for segment in payload.segments:
x = decode_string(segment.x)
y = decode_string(segment.y)
yerr = decode_string(segment.yerr)
seds.append(IrisSed(x=x, y=y, yerr=yerr))
i_stack = IrisStack(seds)
binsize = float(payload.binsize)
statistic = str(payload.statistic)
smooth = payload.smooth == "true"
smooth_binsize = float(payload.smooth_binsize)
logbin = payload.log_bin == "true"
result = sedstacker.sed.stack(i_stack,
binsize,
statistic,
fill='remove',
smooth=smooth,
smooth_binsize=smooth_binsize,
logbin=logbin)
payload.segments[0].x = encode_string(result.x)
payload.segments[0].y = encode_string(result.y)
payload.segments[0].yerr = encode_string(result.yerr)
payload.segments[0].counts = encode_string(result.counts)
payload.segments = [payload.segments[0]]
get_dict = payload.get_dict()
reply_success(msg_id, mtype, payload.get_dict())
except Exception, e:
reply_error(msg_id, sedexceptions.SEDException, e, mtype)
return
except Exception:
error(str(capture_exception()))
def test_dictionary_class(self):
obj = DictionaryClass({'foo': 'bar', 'nested': {'myvar': 5}})
self.assertEqual('bar', obj.foo)
self.assertEqual(5, obj.nested.myvar)
self.assertEqual({
'foo': 'bar',
'nested': {
'myvar': 5
}
}, obj.get_dict())
def stack_stack(private_key, sender_id, msg_id, mtype, params,
extra):
try:
info("stack_stack()")
try:
payload = DictionaryClass(params)
seds = []
for segment in payload.segments:
x = decode_string(segment.x)
y = decode_string(segment.y)
yerr = decode_string(segment.yerr)
seds.append(IrisSed(x=x, y=y, yerr=yerr))
i_stack = IrisStack(seds)
binsize = float(payload.binsize)
statistic = str(payload.statistic)
smooth = payload.smooth == "true"
smooth_binsize = float(payload.smooth_binsize)
logbin = payload.log_bin == "true"
result = sedstacker.sed.stack(i_stack, binsize, statistic, fill='remove', smooth=smooth, smooth_binsize=smooth_binsize, logbin=logbin)
payload.segments[0].x = encode_string(result.x)
payload.segments[0].y = encode_string(result.y)
payload.segments[0].yerr = encode_string(result.yerr)
payload.segments[0].counts = encode_string(result.counts)
payload.segments = [payload.segments[0]]
get_dict = payload.get_dict()
reply_success(msg_id, mtype, payload.get_dict())
except Exception, e:
reply_error(msg_id, sedexceptions.SEDException, e, mtype)
return
except Exception:
error(str(capture_exception()))
