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_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_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_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) )