def test_serialize(self): expected_file = helpers.get_data_path( "expected-dmg-dist-total.xml") expected_text = open(expected_file, "r").readlines() self.make_dist() self.make_data("no_damage", 1.0, 1.6) self.make_data("slight", 34.8, 18.3) self.make_data("moderate", 64.2, 19.8) self.make_data("extensive", 64.3, 19.7) self.make_data("complete", 64.3, 19.7) try: _, result_xml = tempfile.mkstemp() writer = DmgDistTotalXMLWriter(result_xml, "ebl1", self.damage_states) writer.serialize(self.data) actual_text = open(result_xml, "r").readlines() self.assertEqual(expected_text, actual_text) self.assertTrue(xml.validates_against_xml_schema( result_xml)) finally: os.unlink(result_xml)
def test_no_empty_dist(self): # an empty distribution is not supported by the schema writer = DmgDistTotalXMLWriter("output.xml", "ebl1", self.damage_states) self.assertRaises(RuntimeError, writer.serialize, []) self.assertFalse(os.path.exists("output.xml")) self.assertRaises(RuntimeError, writer.serialize, None) self.assertFalse(os.path.exists("output.xml"))
def export_dmg_dist_total(output, target_dir): """ Export the total damage distribution identified by the given output to the `target_dir`. :param output: db output record which identifies the distribution. :type output: :py:class:`openquake.db.models.Output` :param target_dir: destination directory of the exported file. :type target_dir: string """ file_name = "dmg-dist-total-%s.xml" % output.oq_job.id file_path = os.path.join(target_dir, file_name) ddt = models.DmgDistTotal.objects.get(output=output) writer = DmgDistTotalXMLWriter( file_path, ddt.end_branch_label, ddt.dmg_states) data = models.DmgDistTotalData.objects.filter( dmg_dist_total=ddt) writer.serialize(data) return [file_path]