|
1 | | -from parcels import AdvectionRK4, FieldSet, ParticleSet, ScipyParticle, JITParticle |
2 | | -import numpy as np |
3 | 1 | from datetime import timedelta |
4 | 2 |
|
| 3 | +import numpy as np |
| 4 | + |
| 5 | +from parcels import AdvectionRK4, FieldSet, JITParticle, ParticleSet, ScipyParticle |
| 6 | + |
5 | 7 |
|
6 | 8 | class ParticleExecutionJIT: |
7 | 9 | def setup(self): |
@@ -31,14 +33,10 @@ def setup(self): |
31 | 33 | self.pset.execute(AdvectionRK4, runtime=0, dt=timedelta(seconds=30)) |
32 | 34 |
|
33 | 35 | def time_run_single_timestep(self): |
34 | | - self.pset.execute( |
35 | | - AdvectionRK4, runtime=timedelta(seconds=1 * 30), dt=timedelta(seconds=30) |
36 | | - ) |
| 36 | + self.pset.execute(AdvectionRK4, runtime=timedelta(seconds=1 * 30), dt=timedelta(seconds=30)) |
37 | 37 |
|
38 | 38 | def time_run_many_timesteps(self): |
39 | | - self.pset.execute( |
40 | | - AdvectionRK4, runtime=timedelta(seconds=100 * 30), dt=timedelta(seconds=30) |
41 | | - ) |
| 39 | + self.pset.execute(AdvectionRK4, runtime=timedelta(seconds=100 * 30), dt=timedelta(seconds=30)) |
42 | 40 |
|
43 | 41 |
|
44 | 42 | class ParticleExecutionScipy: |
@@ -67,11 +65,7 @@ def setup(self): |
67 | 65 | ) |
68 | 66 |
|
69 | 67 | def time_run_single_timestep(self): |
70 | | - self.pset.execute( |
71 | | - AdvectionRK4, runtime=timedelta(seconds=1 * 30), dt=timedelta(seconds=30) |
72 | | - ) |
| 68 | + self.pset.execute(AdvectionRK4, runtime=timedelta(seconds=1 * 30), dt=timedelta(seconds=30)) |
73 | 69 |
|
74 | 70 | def time_run_many_timesteps(self): |
75 | | - self.pset.execute( |
76 | | - AdvectionRK4, runtime=timedelta(seconds=100 * 30), dt=timedelta(seconds=30) |
77 | | - ) |
| 71 | + self.pset.execute(AdvectionRK4, runtime=timedelta(seconds=100 * 30), dt=timedelta(seconds=30)) |
0 commit comments