def optimize_source_model(input_path, area_src_disc, output_path):
"""
Read a source model and generate a new one, converting area sources to
point sources.
"""
area_src_disc = float(area_src_disc)
source.optimize_source_model(input_path, area_src_disc, output_path)
engine.complain_and_exit(output_path)
def optimize_source_model(input_path, area_src_disc, output_path):
"""
Read a source model and generate a new one, converting area sources to
point sources.
"""
area_src_disc = float(area_src_disc)
source.optimize_source_model(input_path, area_src_disc,
output_path)
engine.complain_and_exit(output_path)
def test_optimize_source_model(self):
in_file = helpers.touch(content=self.input_source_model)
out_file = helpers.touch(content=self.output_source_model)
area_src_disc = 50.0
try:
source_input.optimize_source_model(in_file, area_src_disc,
out_file)
expected = ElementTree.tostring(
ElementTree.XML(self.output_source_model))
actual = ElementTree.tostring(
ElementTree.XML(open(out_file).read()))
self.assertEqual(expected, actual)
finally:
os.unlink(in_file)
os.unlink(out_file)
def test_optimize_source_model(self):
in_file = helpers.touch(content=self.input_source_model)
out_file = helpers.touch(content=self.output_source_model)
area_src_disc = 50.0
try:
source_input.optimize_source_model(in_file, area_src_disc,
out_file)
expected = ElementTree.tostring(
ElementTree.XML(self.output_source_model))
actual = ElementTree.tostring(
ElementTree.XML(open(out_file).read()))
self.assertEqual(expected, actual)
finally:
os.unlink(in_file)
os.unlink(out_file)
def test_optimize_source_model(self):
in_file = helpers.touch(content=self.input_source_model)
out_file = helpers.touch(content=self.output_source_model)
area_src_disc = 50.0
try:
source_input.optimize_source_model(in_file, area_src_disc,
out_file)
expected = etree.tostring(
etree.parse(StringIO.StringIO(self.output_source_model)),
pretty_print=True
)
actual = etree.tostring(etree.parse(out_file), pretty_print=True)
self.assertEqual(expected, actual)
finally:
os.unlink(in_file)
os.unlink(out_file)
