def test14(self):
instance = QGmodel()
instance.parameters.boundary_south = "interface"
instance.commit_parameters()
b1, b2, b3, b4 = instance.get_boundary_conditions()
self.assertEqual(b3, "interface")
instance.stop()
def test1(self):
instance = QGmodel()
instance.evolve_model(1 | units.hour)
self.assertEquals(instance.get_name_of_current_state(), 'EVOLVED')
psi = instance.grid.psi
self.assertEquals(instance.get_name_of_current_state(), 'RUN')
instance.stop()
def test3(self):
instance = QGmodel(redirection="none")
dt = 0.123 | units.hour
instance.parameters.dt = dt
instance.evolve_model(2 * dt)
self.assertEqual(instance.model_time, 2 * dt)
instance.stop()
def test2(self):
instance = QGmodel(redirection="none")
instance.grid[0, 0, 0].psi
self.assertEquals(instance.get_name_of_current_state(), 'EDIT')
instance.initialize_grid()
self.assertEquals(instance.get_name_of_current_state(), 'RUN')
instance.stop()
def test18(self):
instance = QGmodel()
dx = instance.parameters.dx
dy = instance.parameters.dy
dx1, dy1 = instance.grid.cellsize()
self.assertEqual(dx, dx1)
self.assertEqual(dy, dy1)
instance.stop()
def test10(self):
instance = QGmodel()
org = instance.boundaries(1)
self.assertEquals(org.psi.number, 0.)
self.assertEquals(org.dpsi_dt.number, 0.)
shape = org.shape
copy = org.copy()
instance.stop()
def test19(self):
instance = QGmodel()
cellsize = instance.grid.cellsize()
self.assertEquals(instance.get_name_of_current_state(), 'EDIT')
instance.parameters.boundary_west = "interface"
self.assertEquals(instance.get_name_of_current_state(),
'CHANGE_PARAMETERS_EDIT')
instance.recommit_parameters()
self.assertEquals(instance.get_name_of_current_state(), 'EDIT')
instance.stop()
def test23(self):
instance = QGmodel(redirection="none")
Lx = instance.parameters.Lx / 2
instance.parameters.Lx = Lx
instance2 = QGmodel(redirection="none")
print(instance.parameters.copy())
instance2.parameters.reset_from_memento(instance.parameters)
self.assertEqual(instance2.parameters.Lx, Lx)
instance.stop()
instance2.stop()
def test24(self):
instance = QGmodel()
instance.parameters.timestep_method = "euler"
instance.parameters.raw_parameter = 0.123
instance.commit_parameters()
b1 = instance.get_timestep_method()
self.assertEqual(b1, "euler")
b1 = instance.get_raw_alpha()
self.assertEqual(b1, 0.123)
instance.stop()
def test0(self):
instance = QGmodel()
self.assertEquals(instance.get_name_of_current_state(),
'UNINITIALIZED')
instance.initialize_code()
self.assertEquals(instance.get_name_of_current_state(), 'INITIALIZED')
instance.commit_parameters()
self.assertEquals(instance.get_name_of_current_state(), 'EDIT')
instance.evolve_model(1 | units.hour)
self.assertEquals(instance.get_name_of_current_state(), 'EVOLVED')
psi = instance.grid.psi
self.assertEquals(instance.get_name_of_current_state(), 'RUN')
instance.stop()
def test16(self):
instance = QGmodel()
dx = instance.parameters.dx
instance.evolve_model(1 | units.hour)
x = instance.grid[:, 10, 0].x
y = instance.grid[:, 10, 0].y
dpsi1 = instance.grid[:, 10, 0].dpsi_dt
psi2, dpsi2 = instance.get_psi_state_at_point(0. * x + dx, x, y)
self.assertAlmostEquals(dpsi1, dpsi2)
instance.stop()
def test8(self):
instance = QGmodel(redirection="none")
Lx2 = instance.parameters.Lx / 2
instance.parameters.Ly = Lx2
Nx, Ny, Nm = instance.grid.shape
s = instance.boundaries(1).shape
self.assertEqual(s, (2, Ny + 2, 1))
s = instance.boundaries(2).shape
self.assertEqual(s, (2, Ny + 2, 1))
s = instance.boundaries(3).shape
self.assertEqual(s, (Nx + 2, 2, 1))
s = instance.boundaries(4).shape
self.assertEqual(s, (Nx + 2, 2, 1))
instance.stop()
def test20(self):
instance = QGmodel()
cellsize = instance.grid.cellsize()
self.assertEquals(instance.get_name_of_current_state(), 'EDIT')
a = instance.parameters.robert_asselin_alpha
instance.parameters.robert_asselin_alpha = 12345.
self.assertEquals(instance.get_name_of_current_state(),
'CHANGE_PARAMETERS_EDIT')
self.assertEqual(instance.parameters.robert_asselin_alpha, 12345.)
instance.cleanup_code()
self.assertEquals(instance.get_name_of_current_state(), 'END')
instance.initialize_code()
self.assertEqual(instance.parameters.robert_asselin_alpha, a)
self.assertEquals(instance.get_name_of_current_state(), 'INITIALIZED')
instance.stop()
def test6(self):
instance = QGmodel()
Nx, Ny, Nm = instance.grid.shape
self.assertEqual(401, Nx)
minx, maxx, miny, maxy, minz, maxz = instance.get_index_range_inclusive(
)
self.assertEqual(minx, 1)
self.assertEqual(miny, 1)
self.assertEqual(minz, 1)
self.assertEqual(maxx, Nx)
self.assertEqual(maxy, Ny)
self.assertEqual(maxz, Nm)
instance.stop()
def test17(self):
instance = QGmodel()
self.assertEqual(instance.parameters.position_x, 0. | units.m)
self.assertEqual(instance.parameters.position_y, 0. | units.m)
instance.parameters.position_y = 123456. | units.m
instance.parameters.position_x = 543216. | units.m
self.assertEqual(instance.parameters.position_y, 123456. | units.m)
self.assertEqual(instance.parameters.position_x, 543216. | units.m)
x = instance.grid[0, 0, 0].x
y = instance.grid[0, 0, 0].y
self.assertEqual(y, 123456. | units.m)
self.assertEqual(x, 543216. | units.m)
i, j = instance.get_index_of_position(x, y)
self.assertEqual(i, 1)
self.assertEqual(j, 1)
instance.stop()
def test15(self):
instance = QGmodel(redirection="none")
dx = instance.parameters.dx
Lx = instance.parameters.Lx
instance.grid[1:-1, 1:-1, 0].psi = 1. | units.m**2 / units.s
psi, dpsi = instance.get_psi_state_at_point(dx / 2, Lx / 2., Lx / 2.)
self.assertEqual(psi, 1. | units.m**2 / units.s)
psi, dpsi = instance.get_psi_state_at_point(dx, Lx / 2., Lx / 2.)
self.assertEqual(psi, 1. | units.m**2 / units.s)
psi, dpsi = instance.get_psi_state_at_point(2 * dx, Lx / 2., Lx / 2.)
self.assertEqual(psi, 1. | units.m**2 / units.s)
psi, dpsi = instance.get_psi_state_at_point(dx, dx / 2., Lx / 2.)
self.assertEqual(psi, 0.5 | units.m**2 / units.s)
psi, dpsi = instance.get_psi_state_at_point(dx, dx / 2., dx / 2.)
self.assertEqual(psi, 0.25 | units.m**2 / units.s)
instance.stop()
def test11(self):
instance = QGmodel(redirection="none")
instance.parameters.Ly = instance.parameters.Ly / 8
instance.parameters.Lx = instance.parameters.Lx / 8
for i in [1, 2, 3, 4]:
org = instance.boundaries(i)
shape = org.shape
copy = org.copy()
channel = copy.new_channel_to(org)
copy.psi = numpy.random.random(shape) | units.m**2 / units.s
copy.dpsi_dt = numpy.random.random(shape) | units.m**2 / units.s**2
channel.copy_attributes(["psi", "dpsi_dt"])
self.assertEquals(copy.psi, instance.boundaries(i).psi)
self.assertEquals(copy.dpsi_dt, instance.boundaries(i).dpsi_dt)
instance.stop()
def test5(self):
instance = QGmodel(redirection="none")
instance.before_get_parameter()
dt = instance.parameters.dt
instance.evolve_model(dt)
dpsi_dt1 = instance.grid.dpsi_dt
psi1 = instance.grid.psi
instance.evolve_model(2 * dt)
psi2 = instance.grid.psi
dpsi_dt2 = instance.grid.dpsi_dt
dpsi_estimate = ((psi2 - psi1) / dt).mean()
dpsi = 0.5 * (dpsi_dt1 + dpsi_dt2).mean()
self.assertTrue(abs(dpsi - dpsi_estimate) / dpsi < 0.1)
instance.stop()
def test7(self):
instance = QGmodel()
Nx, Ny, Nm = instance.grid.shape
self.assertEqual(401, Nx)
minx, maxx, miny, maxy, minz, maxz = instance.get_boundary_index_range_inclusive(
1)
self.assertEqual(minx, 0)
self.assertEqual(miny, 0)
self.assertEqual(minz, 1)
self.assertEqual(maxx, 1)
self.assertEqual(maxy, Ny + 1)
self.assertEqual(maxz, Nm)
minx, maxx, miny, maxy, minz, maxz = instance.get_boundary_index_range_inclusive(
2)
self.assertEqual(minx, Nx)
self.assertEqual(miny, 0)
self.assertEqual(minz, 1)
self.assertEqual(maxx, Nx + 1)
self.assertEqual(maxy, Ny + 1)
self.assertEqual(maxz, Nm)
minx, maxx, miny, maxy, minz, maxz = instance.get_boundary_index_range_inclusive(
3)
self.assertEqual(minx, 0)
self.assertEqual(miny, 0)
self.assertEqual(minz, 1)
self.assertEqual(maxx, Nx + 1)
self.assertEqual(maxy, 1)
self.assertEqual(maxz, Nm)
minx, maxx, miny, maxy, minz, maxz = instance.get_boundary_index_range_inclusive(
4)
self.assertEqual(minx, 0)
self.assertEqual(miny, Ny)
self.assertEqual(minz, 1)
self.assertEqual(maxx, Nx + 1)
self.assertEqual(maxy, Ny + 1)
self.assertEqual(maxz, Nm)
instance.stop()
def test12(self):
instance = QGmodel()
instance.parameters.Ly = instance.parameters.Ly / 16
instance.parameters.Lx = instance.parameters.Lx / 16
Nx, Ny, Nm = instance.grid.shape
ix = numpy.arange(Nx) + 1
iy = numpy.arange(Ny) + 1
onesx = numpy.ones(Nx)
onesy = numpy.ones(Ny)
i1 = numpy.outer(ix, onesx).flatten()
j1 = numpy.outer(onesy, iy).flatten()
x = instance.grid.x.flatten()
y = instance.grid.y.flatten()
i, j = instance.get_index_of_position(x, y)
self.assertEquals(i - i1, 0.)
self.assertEquals(j - j1, 0.)
instance.stop()
def test4(self):
instance = QGmodel(redirection="none")
instance.grid[0, 0, 0].dpsi_dt
self.assertEquals(instance.get_name_of_current_state(), 'RUN')
instance.stop()
def test21(self):
instance = QGmodel()
self.assertFalse(instance.parameters.interface_wind)
instance.parameters.interface_wind = True
self.assertTrue(instance.parameters.interface_wind)
instance.stop()
def test9(self):
instance = QGmodel()
org = instance.grid
copy = org.copy()
instance.stop()