def test_csv_to_xml(self): """ Tests the conversion of csv to xml """ # Convert csv file csv_to_xml(self.input_csv, self.output_xml) # Load in both reference and exported xml files parser1 = SiteModelParser(self.output_xml) model1 = list(parser1.parse()) parser2 = SiteModelParser(self.reference_xml) model2 = list(parser2.parse()) # Test comparison self._compare_site_models(model1, model2) # Cleanup os.remove(self.output_xml)
def xml_to_csv(input_xml, output_csv, geo_precision="%12.6f", site_precision="%8.2f"): """ Converts the xml to csv format """ parser = SiteModelParser(input_xml) sites = list(parser.parse()) f = open(output_csv, "w") target_string = ",".join([geo_precision, geo_precision, site_precision, "%s", site_precision, site_precision]) for site in sites: locn = wkt.loads(site.wkt) print >> f, target_string % (locn.x, locn.y, site.vs30, site.vs30_type, site.z1pt0, site.z2pt5) f.close()