def test_seed_shapefile(self):
o = OceanDrift(loglevel=20)
o.seed_from_shapefile(o.test_data_folder() +
'shapefile_spawning_areas/Torsk.shp',
number=100, layername=None,
featurenum=[2, 4], time=datetime.now())
self.assertEqual(len(o.elements_scheduled), 100)
o.seed_from_shapefile(o.test_data_folder() +
'shapefile_spawning_areas/Torsk.shp',
number=300, layername=None,
featurenum=None, time=datetime.now())
self.assertEqual(len(o.elements_scheduled), 400)
def test_seed_shapefile(self):
o = OceanDrift(loglevel=20)
o.seed_from_shapefile(o.test_data_folder() +
'shapefile_spawning_areas/Torsk.shp',
number=100, layername=None,
featurenum=[2, 4], time=datetime.now())
self.assertEqual(len(o.elements_scheduled), 100)
o.seed_from_shapefile(o.test_data_folder() +
'shapefile_spawning_areas/Torsk.shp',
number=300, layername=None,
featurenum=None, time=datetime.now())
self.assertEqual(len(o.elements_scheduled), 400)
#!/usr/bin/env python
"""
Seeding from shapefile
==================================
"""
from datetime import datetime
from opendrift.models.oceandrift import OceanDrift
o = OceanDrift(loglevel=20) # Set loglevel to 0 for debug information
#%%
# Seed oil particles within contours from shapefile
o.seed_from_shapefile(o.test_data_folder() +
'shapefile_spawning_areas/Torsk.shp',
number=2000,
layername=None,
featurenum=[2, 4],
time=datetime.now())
o.fallback_values['x_wind'] = -4 # Constant wind drift
o.fallback_values['y_wind'] = 8
o.set_config('drift:wind_uncertainty', 4) # Adding some diffusion
#%%
# Running model
o.run(steps=50, time_step=3600)
#%%
# Print and plot results
print(o)
o.plot(fast=True)
