def test_count_equal(self):
layer = Layer(10, 10)
self.assertEqual(100, layer.countEqual(0))
self.assertEqual(0, layer.countEqual(1))
layer[3, 3] = 3
self.assertEqual(1, layer.countEqual(3))
def test_matching(self):
d = 10
layer = Layer(d, d)
for i in range(d):
layer[i, i] = 10
cell_list = layer.cellsEqual(1)
self.assertEqual(cell_list, [])
cell_list = layer.cellsEqual(10)
self.assertEqual(cell_list, [(i, i) for i in range(d)])
def test_get_barrier(self):
layer = Layer(10, 10)
self.assertFalse(layer.leftBarrier(5, 5))
self.assertFalse(layer.bottomBarrier(5, 5))
layer.addIJBarrier([(1, 1), (2, 1), (2, 2)])
self.assertTrue(layer.bottomBarrier(1, 1))
self.assertTrue(layer.leftBarrier(2, 1))
def test_add_cell(self):
layer = Layer(10, 10)
with self.assertRaises(ValueError):
layer[100, 100] = 199
with self.assertRaises(ValueError):
layer[100, "X"] = 199
with self.assertRaises(ValueError):
layer[100] = 199
layer[5, 5] = 88
self.assertEqual(layer[5, 5], 88)
def test_layer(self):
region = EclRegion(self.grid, False)
layer = Layer(self.grid.getNX(), self.grid.getNY() + 1)
with self.assertRaises(ValueError):
region.selectFromLayer(layer, 0, 1)
layer = Layer(self.grid.getNX(), self.grid.getNY())
layer[0, 0] = 1
layer[1, 1] = 1
layer[2, 2] = 1
with self.assertRaises(ValueError):
region.selectFromLayer(layer, -1, 1)
with self.assertRaises(ValueError):
region.selectFromLayer(layer, self.grid.getNZ(), 1)
region.selectFromLayer(layer, 0, 2)
glist = region.getGlobalList()
self.assertEqual(0, len(glist))
region.selectFromLayer(layer, 0, 1)
glist = region.getGlobalList()
self.assertEqual(3, len(glist))
def test_add_polyline_barrier(self):
d = 10
layer = Layer(d, d)
grid = EclGrid.createRectangular((d, d, 1), (1, 1, 1))
pl = CPolyline(init_points=[(0, 0), (d / 2, d / 2), (d, d)])
layer.addPolylineBarrier(pl, grid, 0)
for i in range(d):
self.assertTrue(layer.bottomBarrier(i, i))
if i < (d - 1):
self.assertTrue(layer.leftBarrier(i + 1, i))
def test_contact(self):
nx = 20
ny = 10
layer = Layer(nx, ny)
grid = EclGrid.createRectangular((nx, ny, 1), (1, 1, 1))
with self.assertRaises(IndexError):
layer.cellContact((-1, 0), (1, 1))
with self.assertRaises(IndexError):
layer.cellContact((20, 0), (1, 1))
self.assertFalse(layer.cellContact((0, 0), (2, 0)))
self.assertFalse(layer.cellContact((1, 0), (1, 0)))
self.assertTrue(layer.cellContact((0, 0), (1, 0)))
self.assertTrue(layer.cellContact((1, 0), (0, 0)))
self.assertTrue(layer.cellContact((0, 0), (0, 1)))
self.assertTrue(layer.cellContact((0, 1), (0, 0)))
self.assertFalse(layer.cellContact((0, 0), (1, 1)))
self.assertFalse(layer.cellContact((1, 1), (0, 0)))
self.assertTrue(layer.cellContact((4, 0), (5, 0)))
self.assertTrue(layer.cellContact((0, 4), (0, 5)))
with TestAreaContext("Layer/barrier"):
with open("faults.grdecl", "w") as f:
f.write("FAULTS\n")
f.write("\'FX\' 5 5 1 10 1 1 'X' /\n")
f.write("\'FY\' 1 10 5 5 1 1 'Y' /\n")
f.write("/")
faults = FaultCollection(grid, "faults.grdecl")
layer.addFaultBarrier(faults["FX"], 0)
self.assertFalse(layer.cellContact((4, 0), (5, 0)))
layer.addFaultBarrier(faults["FY"], 0)
self.assertFalse(layer.cellContact((0, 4), (0, 5)))
self.assertFalse(layer.cellContact((9, 4), (9, 5)))
self.assertTrue(layer.cellContact((10, 4), (10, 5)))
def test_create_layer(self):
layer = Layer(10, 10)
self.assertTrue(isinstance(layer, Layer))
def test_assign(self):
layer = Layer(10, 5)
self.assertEqual(layer.cellSum(), 0)
layer.assign(10)
self.assertEqual(layer.cellSum(), 500)
def test_active(self):
d = 10
layer = Layer(d, d)
with self.assertRaises(ValueError):
layer.activeCell(d + 1, d + 2)
self.assertTrue(layer.activeCell(1, 2))
grid = EclGrid.createRectangular((d, d + 1, 1), (1, 1, 1))
with self.assertRaises(ValueError):
layer.updateActive(grid, 0)
grid = EclGrid.createRectangular((d, d, 1), (1, 1, 1))
with self.assertRaises(ValueError):
layer.updateActive(grid, 10)
actnum = IntVector(initial_size=d * d * 1, default_value=1)
actnum[0] = 0
grid = EclGrid.createRectangular((d, d, 1), (1, 1, 1), actnum=actnum)
layer.updateActive(grid, 0)
self.assertTrue(layer.activeCell(1, 2))
self.assertFalse(layer.activeCell(0, 0))
def test_update_connected(self):
nx = 10
ny = 10
layer = Layer(nx, ny)
layer[0, 0] = 100
self.assertEqual(layer[0, 0], 100)
layer.clearCells()
self.assertEqual(layer[0, 0], 0)
self.assertEqual(layer.cellSum(), 0)
with self.assertRaises(ValueError):
layer.updateConnected((10, 10), 10)
layer[0, 0] = 77
with self.assertRaises(ValueError):
layer.updateConnected((0, 0), 10, org_value=0)
layer.updateConnected((0, 0), 10)
self.assertEqual(10, layer.cellSum())
layer[0, 0] = 0
layer.updateConnected((0, 0), 3)
self.assertEqual(nx * ny * 3, layer.cellSum())
layer.addIJBarrier([(5, 0), (5, 10)])
layer.clearCells()
self.assertEqual(0, layer.cellSum())
layer.updateConnected((0, 0), 1)
self.assertEqual(50, layer.cellSum())
self.assertEqual(layer[4, 0], 1)
self.assertEqual(layer[5, 0], 0)
layer = Layer(nx, ny)
layer.addIJBarrier([(5, 0), (5, 5)])
layer.updateConnected((0, 0), 1)
self.assertEqual(100, layer.cellSum())
def test_contact2(self):
nx = 10
ny = 10
layer = Layer(nx, ny)
grid = EclGrid.createRectangular((nx, ny, 1), (1, 1, 1))
# Too short
with self.assertRaises(ValueError):
layer.addIJBarrier([(1, 5)])
# Out of range
with self.assertRaises(ValueError):
layer.addIJBarrier([(10, 15), (5, 5)])
# Out of range
with self.assertRaises(ValueError):
layer.addIJBarrier([(7, 7), (-5, 5)])
# Must have either i1 == i2 or j1 == j2
with self.assertRaises(ValueError):
layer.addIJBarrier([(7, 8), (6, 5)])
p1 = (0, 4)
p2 = (0, 5)
self.assertTrue(layer.cellContact(p1, p2))
layer.addIJBarrier([(0, 5), (nx, 5)])
self.assertFalse(layer.cellContact(p1, p2))
def test_fault_barrier(self):
nx = 120
ny = 60
nz = 43
grid = EclGrid.createRectangular((nx, ny, nz), (1, 1, 1))
with TestAreaContext("python/faults/line_order"):
with open("faults.grdecl", "w") as f:
f.write("""FAULTS
\'F\' 105 107 50 50 1 43 \'Y\' /
\'F\' 108 108 50 50 1 43 \'X\' /
\'F\' 108 108 50 50 22 43 \'Y\' /
\'F\' 109 109 49 49 1 43 \'Y\' /
\'F\' 110 110 49 49 1 43 \'X\' /
\'F\' 111 111 48 48 1 43 \'Y\' /
/
""")
with open("faults.grdecl") as f:
faults = FaultCollection(grid, "faults.grdecl")
# Fault layout:
#
# +---+---+---+---+
# |
# +---+ +
# |
# +---+
fault = faults["F"]
layer = Layer(nx, ny)
fault_pairs = [((104, 49), (104, 50)), ((105, 49), (105, 50)),
((106, 49), (106, 50)), ((107, 49), (108, 49)),
((107, 49), (107, 50)), ((108, 48), (108, 49)),
((109, 48), (110, 48)), ((110, 47), (110, 48))]
gap_pair = ((109, 48), (109, 49))
for p1, p2 in fault_pairs:
self.assertTrue(layer.cellContact(p1, p2))
p1, p2 = gap_pair
self.assertTrue(layer.cellContact(p1, p2))
layer.addFaultBarrier(fault, 30, link_segments=False)
for p1, p2 in fault_pairs:
self.assertFalse(layer.cellContact(p1, p2))
p1, p2 = gap_pair
self.assertTrue(layer.cellContact(p1, p2))
layer.addFaultBarrier(fault, 30)
p1, p2 = gap_pair
self.assertFalse(layer.cellContact(p1, p2))