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)