def test_write_linestrings_yaml(self):
fpath_in = os.path.join(self.dir_test, 'flowlines.yaml')
lines = schism_linestring.read_linestrings(fpath_in)
fpath_out = 'test_linestrings.yaml'
try:
schism_linestring.write_linestrings(fpath_out, lines)
lines_readback = schism_linestring.read_linestrings(fpath_out)
self.assertEqual([l.prop['name'] for l in lines],
[l.prop['name'] for l in lines_readback])
self.assertEqual([list(l.coords) for l in lines],
[list(l.coords) for l in lines_readback])
finally:
if os.path.exists(fpath_out):
os.remove(fpath_out)
def main():
""" A main function to convert polygon files
"""
parser = create_arg_parser()
args = parser.parse_args()
if args.input.endswith('.yaml'):
linestrings = read_linestrings(args.input)
if args.output.endswith('.shp'):
write_linestrings(args.output, linestrings)
else:
raise ValueError("Not supported output file type")
elif args.input.endswith('.shp'):
linestrings = read_linestrings(args.input)
write_linestrings(args.output, linestrings)
else:
raise ValueError("Not supported input file type")
def test_read_linestrings_shapefile(self):
fpath = os.path.join(self.dir_test, 'test_linestrings.shp')
lines = schism_linestring.read_linestrings(fpath)
self.assertEqual([l.prop['name'] for l in lines],
['ocean', 'mixed1', 'middle'])
self.assertEqual([list(l.coords) for l in lines],
[[(41.0, 69.0), (41.0, 101.0)],
[(101.0, 75.0), (69.0, 75.0)],
[(45.0, 5.0), (45.0, 25.0)]])
def test_write_linestrings_shp(self):
fpath_in = os.path.join(self.dir_test, 'flowlines.yaml')
lines = schism_linestring.read_linestrings(fpath_in)
fpath_out = 'test_linestrings.shp'
try:
schism_linestring.write_linestrings(fpath_out, lines)
finally:
files_to_remove = glob.glob(os.path.splitext(fpath_out)[0] + '.*')
for filename in files_to_remove:
if os.path.exists(filename):
os.remove(filename)