def test_loss_curve_writer_creation(self):
# XML writers
writer = risk_output.create_loss_curve_writer(
None, ['xml'], "fakepath.xml", "loss_ratio")
self.assertEqual(type(writer), risk_output.LossRatioCurveXMLWriter)
writer = risk_output.create_loss_curve_writer(
None, ['xml'], "fakepath.xml", "loss")
self.assertEqual(type(writer), risk_output.LossCurveXMLWriter)
# database writers
writer = risk_output.create_loss_curve_writer(
1, ['db'], "fakepath.xml", "loss_ratio")
self.assertEqual(writer, None)
writer = risk_output.create_loss_curve_writer(
1, ['db'], "fakepath.xml", "loss")
self.assertEqual(type(writer), risk_output.LossCurveDBWriter)
def _serialize(self, block_id, **kwargs):
"""
Build filename/paths for serializing and call _serialize
Return the list of filenames. The list will be empty if nothing was
actually serialized.
"""
if kwargs['curve_mode'] == 'loss_ratio':
serialize_filename = "%s-block-#%s-block#%s.xml" % (
self.calc_proxy.params["LOSS_CURVES_OUTPUT_PREFIX"],
self.calc_proxy.job_id,
block_id)
elif kwargs['curve_mode'] == 'loss':
serialize_filename = "%s-loss-block-#%s-block#%s.xml" % (
self.calc_proxy.params["LOSS_CURVES_OUTPUT_PREFIX"],
self.calc_proxy.job_id,
block_id)
serialize_path = os.path.join(self.calc_proxy.base_path,
self.calc_proxy.params['OUTPUT_DIR'],
serialize_filename)
LOG.debug("Serializing %s" % kwargs['curve_mode'])
writer = risk_output.create_loss_curve_writer(
self.calc_proxy.job_id, self.calc_proxy.serialize_results_to,
serialize_path, kwargs['curve_mode'])
if writer:
writer.serialize(kwargs['curves'])
return [serialize_path]
else:
return []
def _serialize(self, block_id, **kwargs):
"""
Build filename/paths for serializing and call _serialize
Return the list of filenames. The list will be empty if nothing was
actually serialized.
"""
if kwargs['curve_mode'] == 'loss_ratio':
serialize_filename = "%s-block-#%s-block#%s.xml" % (
self.job_ctxt.params["LOSS_CURVES_OUTPUT_PREFIX"],
self.job_ctxt.job_id, block_id)
elif kwargs['curve_mode'] == 'loss':
serialize_filename = "%s-loss-block-#%s-block#%s.xml" % (
self.job_ctxt.params["LOSS_CURVES_OUTPUT_PREFIX"],
self.job_ctxt.job_id, block_id)
serialize_path = os.path.join(self.job_ctxt.base_path,
self.job_ctxt.params['OUTPUT_DIR'],
serialize_filename)
LOG.debug("Serializing %s" % kwargs['curve_mode'])
writer = risk_output.create_loss_curve_writer(
self.job_ctxt.job_id, self.job_ctxt.serialize_results_to,
serialize_path, kwargs['curve_mode'])
if writer:
writer.serialize(kwargs['curves'])
return [serialize_path]
else:
return []
def test_loss_curve_writer_creation(self):
# XML writers
writer = risk_output.create_loss_curve_writer(None, ['xml'],
"fakepath.xml",
"loss_ratio")
self.assertEqual(type(writer), risk_output.LossRatioCurveXMLWriter)
writer = risk_output.create_loss_curve_writer(None, ['xml'],
"fakepath.xml", "loss")
self.assertEqual(type(writer), risk_output.LossCurveXMLWriter)
# database writers
writer = risk_output.create_loss_curve_writer(1, ['db'],
"fakepath.xml",
"loss_ratio")
self.assertEqual(writer, None)
writer = risk_output.create_loss_curve_writer(1, ['db'],
"fakepath.xml", "loss")
self.assertEqual(type(writer), risk_output.LossCurveDBWriter)
