def test_local_surface(self):
with ErtTestContext('python/enkf/data/local_config', self.surf_config) as test_context:
main = test_context.getErt()
local_config = main.getLocalConfig()
# Creating dataset
data_scale = local_config.createDataset('DATA_SCALE')
surf = self._small_surf()
ps = surf.getPointset()
geo_reg = GeoRegion(ps)
data_scale.addSurface('TOP', geo_reg)
self.assertEqual(1, len(data_scale))
def test_raises(self):
surface = self.small_surface()
pointset = GeoPointset.fromSurface(surface)
georegion = GeoRegion(pointset)
with self.assertRaises(ValueError):
georegion.select_above(((2, ), (1, 3)))
with self.assertRaises(ValueError):
georegion.select_above((('not-a-number', 2), (1, 3)))
def test_select_halfspace(self):
surface = self.small_surface()
pointset = GeoPointset.fromSurface(surface)
georegion = GeoRegion(pointset)
self.assertEqual(0, len(georegion))
line = [(-0.1, 2.0), (1.9, 8.1)]
picked = 118
georegion.select_above(line)
self.assertEqual(picked, len(georegion))
georegion.deselect_above(line)
self.assertEqual(0, len(georegion))
georegion.select_below(line)
self.assertEqual(len(surface) - picked, len(georegion))
georegion.deselect_below(line)
self.assertEqual(0, len(georegion))
georegion.select_above(line)
georegion.select_below(line)
self.assertEqual(len(surface), len(georegion))
georegion.deselect_above(line)
georegion.deselect_below(line)
self.assertEqual(0, len(georegion))
def test_init(self):
pointset = GeoPointset()
georegion = GeoRegion(pointset)
self.assertEqual(0, len(georegion))
def test_select_polygon(self):
surface = self.small_surface()
pointset = GeoPointset.fromSurface(surface)
georegion = GeoRegion(pointset)
self.assertEqual(0, len(georegion))
points = [(-0.1, 2.0), (1.9, 8.1), (6.1, 8.1), (9.1, 5), (7.1, 0.9)]
polygon = CPolyline(name='test_polygon', init_points=points)
picked = 52 # https://www.futilitycloset.com/2013/04/24/picks-theorem/
georegion.select_inside(polygon)
self.assertEqual(picked, len(georegion))
georegion.deselect_inside(polygon)
self.assertEqual(0, len(georegion))
georegion.select_outside(polygon)
self.assertEqual(len(surface) - picked, len(georegion))
georegion.deselect_outside(polygon)
self.assertEqual(0, len(georegion))
georegion.select_inside(polygon)
georegion.select_outside(polygon)
self.assertEqual(len(surface), len(georegion))
georegion.deselect_inside(polygon)
georegion.deselect_outside(polygon)
self.assertEqual(0, len(georegion))
georegion.select_inside(polygon)
self.assertEqual(picked, len(georegion))
internal_square = [(2.5, 2.5), (2.5, 6.5), (6.5, 6.5), (6.5, 2.5)]
georegion.deselect_inside(CPolyline(init_points=internal_square))
self.assertEqual(picked - 4 * 4,
len(georegion)) # internal square is 4x4
def test_repr(self):
pointset = GeoPointset()
georegion = GeoRegion(pointset)
self.assertTrue(repr(georegion).startswith('GeoRegion'))