def test_getitem(self):
srf_path = self.createTestPath(
"local/geometry/surface/valid_ascii.irap")
srf = Surface(srf_path)
gp = GeoPointset.fromSurface(srf)
for i in (561, 1105, 1729, 2465, 2821):
self.assertEqual(gp[i], srf[i])
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_from_surface(self):
srf_path = self.createTestPath(
"local/geometry/surface/valid_ascii.irap")
srf = Surface(srf_path)
gp = GeoPointset.fromSurface(srf)
self.assertEqual(3871, len(srf))
self.assertEqual(len(srf), len(gp))
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))