def test_init_model(self):
"""
Should initialize a new model
"""
shape = (128, 1, 64)
vm = VM(shape=shape)
# should have a created the 3D grid manager
self.assertTrue(hasattr(vm, 'grid'))
self.assertEqual(vm.grid.nx, shape[0])
self.assertEqual(vm.grid.ny, shape[1])
self.assertEqual(vm.grid.nz, shape[2])
# should have default attributes needed by other routines
for attr in [
'sl', 'nr', 'nx', 'ny', 'nz', 'dx', 'dy', 'dz', 'rf', 'jp',
'ij', 'ir', 'r1', 'r2', 'ilyr'
]:
self.assertTrue(hasattr(vm, attr))
# grid should be initialized to zeros
self.assertEqual(vm.sl[0, 0, 0], 0)
# setting certain aliases should set parent
vm.sl[0, 0, 0] = 999
self.assertEqual(vm.grid.values[0, 0, 0], 999)
for attr in ['dx', 'dy', 'dz']:
vm.__setattr__(attr, 999)
self.assertEqual(vm.__getattribute__(attr), 999)
self.assertEqual(vm.grid.__getattribute__(attr), 999)
vm.r1 = (9.99, 8.88, 7.77)
self.assertEqual(vm.r1, (9.99, 8.88, 7.77))
self.assertEqual(vm.grid.origin, (9.99, 8.88, 7.77))
# setting r2 should update grid spacing
nx0, ny0, nz0 = vm.grid.shape[:]
dx0, dy0, dz0 = vm.grid.spacing[:]
vm.r2 = (400, 200, 300)
self.assertEqual(vm.nx, nx0)
self.assertEqual(vm.ny, ny0)
self.assertEqual(vm.nz, nz0)
self.assertNotEqual(vm.dx, dx0)
self.assertNotEqual(vm.dy, dy0)
self.assertNotEqual(vm.dz, dz0)
# extents must be 3D
for attr in ['r1', 'r2']:
with self.assertRaises(ValueError) as context:
vm.__setattr__(attr, (0, 0))
# should not allow certain aliases to be set
for attr in ['nx', 'ny', 'nz', 'nr']:
with self.assertRaises(AttributeError) as context:
vm.__setattr__(attr, 999)
def test_init_model(self):
"""
Should initialize a new model
"""
shape = (128, 1, 64)
vm = VM(shape=shape)
# should have a created the 3D grid manager
self.assertTrue(hasattr(vm, 'grid'))
self.assertEqual(vm.grid.nx, shape[0])
self.assertEqual(vm.grid.ny, shape[1])
self.assertEqual(vm.grid.nz, shape[2])
# should have default attributes needed by other routines
for attr in ['sl', 'nr', 'nx', 'ny', 'nz', 'dx', 'dy', 'dz',
'rf', 'jp', 'ij', 'ir', 'r1', 'r2', 'ilyr']:
self.assertTrue(hasattr(vm, attr))
# grid should be initialized to zeros
self.assertEqual(vm.sl[0, 0, 0], 0)
# setting certain aliases should set parent
vm.sl[0, 0, 0] = 999
self.assertEqual(vm.grid.values[0, 0, 0], 999)
for attr in ['dx', 'dy', 'dz']:
vm.__setattr__(attr, 999)
self.assertEqual(vm.__getattribute__(attr), 999)
self.assertEqual(vm.grid.__getattribute__(attr), 999)
vm.r1 = (9.99, 8.88, 7.77)
self.assertEqual(vm.r1, (9.99, 8.88, 7.77))
self.assertEqual(vm.grid.origin, (9.99, 8.88, 7.77))
# setting r2 should update grid spacing
nx0, ny0, nz0 = vm.grid.shape[:]
dx0, dy0, dz0 = vm.grid.spacing[:]
vm.r2 = (400, 200, 300)
self.assertEqual(vm.nx, nx0)
self.assertEqual(vm.ny, ny0)
self.assertEqual(vm.nz, nz0)
self.assertNotEqual(vm.dx, dx0)
self.assertNotEqual(vm.dy, dy0)
self.assertNotEqual(vm.dz, dz0)
# extents must be 3D
for attr in ['r1', 'r2']:
with self.assertRaises(ValueError) as context:
vm.__setattr__(attr, (0, 0))
# should not allow certain aliases to be set
for attr in ['nx', 'ny', 'nz', 'nr']:
with self.assertRaises(AttributeError) as context:
vm.__setattr__(attr, 999)
def test_r1_r2(self):
"""
Should force reflectors to be the same shape as the grid
"""
shape = (128, 1, 64)
vm = VM(shape=shape)
for attr in ['rf', 'jp', 'ir', 'ij']:
# should allow (nr, nx, ny)-sized arrays
vm.__setattr__(attr, np.ones((1, vm.nx, vm.ny)))
self.assertEqual(vm.nr, 1)
# should not allow other sizes
with self.assertRaises(ValueError) as context:
vm.__setattr__(attr, np.ones((1, 999, vm.ny)))
with self.assertRaises(ValueError) as context:
vm.__setattr__(attr, np.ones((1, vm.nx, 999)))
def test_r1_r2(self):
"""
Should force reflectors to be the same shape as the grid
"""
shape = (128, 1, 64)
vm = VM(shape=shape)
for attr in ['rf', 'jp', 'ir', 'ij']:
# should allow (nr, nx, ny)-sized arrays
vm.__setattr__(attr, np.ones((1, vm.nx, vm.ny)))
self.assertEqual(vm.nr, 1)
# should not allow other sizes
with self.assertRaises(ValueError) as context:
vm.__setattr__(attr, np.ones((1, 999, vm.ny)))
with self.assertRaises(ValueError) as context:
vm.__setattr__(attr, np.ones((1, vm.nx, 999)))