class TestBinifyExcludeEmpty(unittest.TestCase):
def setUp(self):
self.args = [
'tests/test-shapefiles/simple-points.shp',
'tests/test-shapefiles/simple-points-grid-empty.shp',
'--exclude-empty',
'--overwrite',
'--suppress-output',
]
self.b = Binifier(self.args)
self.b.main()
self.driver = ogr.GetDriverByName('ESRI Shapefile')
self.out_shapefile = self.driver.Open(
'tests/test-shapefiles/simple-points-grid-empty.shp', GA_ReadOnly)
self.out_layer = self.out_shapefile.GetLayer()
def test_feature_count(self):
# We must loop because features are marked for deletion, but are still
# considered in layer.GetFeatureCount()
count = 0
another_feature = True
while (another_feature):
feature = self.out_layer.GetNextFeature()
if feature:
count += 1
feature.Destroy()
else:
another_feature = False
self.out_layer.ResetReading()
self.assertEqual(count, 7)
def tearDown(self):
self.out_shapefile.Destroy()
def launch_new_instance():
"""
Launch an instance of Binifier.
This is the entry function of the command-line tool `binify`.
"""
binifier = Binifier()
binifier.main()
def launch_new_instance():
"""
Launch an instance of Binifier.
This is the entry function of the command-line tool `binify`.
"""
binifier = Binifier()
binifier.main()
def setUp(self):
self.args = [
'tests/test-shapefiles/simple-points.shp',
'tests/test-shapefiles/simple-points-grid.shp',
'--overwrite',
'--suppress-output',
]
self.b = Binifier(self.args)
self.b.main()
self.driver = ogr.GetDriverByName('ESRI Shapefile')
self.out_shapefile = self.driver.Open(
'tests/test-shapefiles/simple-points-grid.shp', GA_ReadOnly)
self.out_layer = self.out_shapefile.GetLayer()
class TestBinifySimple(unittest.TestCase):
def setUp(self):
self.args = [
'tests/test-shapefiles/simple-points.shp',
'tests/test-shapefiles/simple-points-grid.shp',
'--overwrite',
'--suppress-output',
]
self.b = Binifier(self.args)
self.b.main()
self.driver = ogr.GetDriverByName('ESRI Shapefile')
self.out_shapefile = self.driver.Open(
'tests/test-shapefiles/simple-points-grid.shp',
GA_ReadOnly
)
self.out_layer = self.out_shapefile.GetLayer()
def test_extent(self):
extent = self.out_layer.GetExtent()
self.assertEqual(extent, (
-0.7789222489017765,
-0.03954617630398981,
-0.3489926778373724,
0.1998517180083852)
)
def test_count(self):
count = 0
another_feature = True
while (another_feature):
feature = self.out_layer.GetNextFeature()
if feature:
count += feature.GetFieldAsInteger('COUNT')
else:
another_feature = False
self.assertEqual(count, 10)
def test_feature_count(self):
self.assertEqual(self.out_layer.GetFeatureCount(), 160)
def tearDown(self):
self.out_shapefile.Destroy()
class TestBinifySimple(unittest.TestCase):
def setUp(self):
self.args = [
'tests/test-shapefiles/simple-points.shp',
'tests/test-shapefiles/simple-points-grid.shp',
'--overwrite',
'--suppress-output',
]
self.b = Binifier(self.args)
self.b.main()
self.driver = ogr.GetDriverByName('ESRI Shapefile')
self.out_shapefile = self.driver.Open(
'tests/test-shapefiles/simple-points-grid.shp',
GA_ReadOnly
)
self.out_layer = self.out_shapefile.GetLayer()
def test_extent(self):
extent = self.out_layer.GetExtent()
self.assertEqual(extent, (
-0.7789222489017765,
-0.03954617630398981,
-0.3489926778373724,
0.1998517180083852)
)
def test_count(self):
count = 0
another_feature = True
while (another_feature):
feature = self.out_layer.GetNextFeature()
if feature:
count += feature.GetFieldAsInteger('COUNT')
else:
another_feature = False
self.assertEqual(count, 10)
def test_feature_count(self):
self.assertEqual(self.out_layer.GetFeatureCount(), 160)
def tearDown(self):
self.out_shapefile.Destroy()
class TestBinifyExcludeEmpty(unittest.TestCase):
def setUp(self):
self.args = [
'tests/test-shapefiles/simple-points.shp',
'tests/test-shapefiles/simple-points-grid-empty.shp',
'--exclude-empty',
'--overwrite',
'--suppress-output',
]
self.b = Binifier(self.args)
self.b.main()
self.driver = ogr.GetDriverByName('ESRI Shapefile')
self.out_shapefile = self.driver.Open(
'tests/test-shapefiles/simple-points-grid-empty.shp',
GA_ReadOnly
)
self.out_layer = self.out_shapefile.GetLayer()
def test_feature_count(self):
# We must loop because features are marked for deletion, but are still
# considered in layer.GetFeatureCount()
count = 0
another_feature = True
while (another_feature):
feature = self.out_layer.GetNextFeature()
if feature:
count += 1
feature.Destroy()
else:
another_feature = False
self.out_layer.ResetReading()
self.assertEqual(count, 7)
def tearDown(self):
self.out_shapefile.Destroy()
def setUp(self):
self.args = [
'tests/test-shapefiles/simple-points.shp',
'tests/test-shapefiles/simple-points-grid.shp',
'--overwrite',
'--suppress-output',
]
self.b = Binifier(self.args)
self.b.main()
self.driver = ogr.GetDriverByName('ESRI Shapefile')
self.out_shapefile = self.driver.Open(
'tests/test-shapefiles/simple-points-grid.shp',
GA_ReadOnly
)
self.out_layer = self.out_shapefile.GetLayer()
