class TestOGRShapeReader(object): def setup(self): self.reader = OGRShapeReader(polygon_file) def test_read_all(self): wkts = list(self.reader.wkts()) eq_(len(wkts), 3) for wkt in wkts: assert wkt.startswith(b'POLYGON ('), 'unexpected WKT: %s' % wkt def test_read_filter(self): wkts = list(self.reader.wkts(where='name = "germany"')) eq_(len(wkts), 2) for wkt in wkts: assert wkt.startswith(b'POLYGON ('), 'unexpected WKT: %s' % wkt def test_read_filter_no_match(self): wkts = list(self.reader.wkts(where='name = "foo"')) eq_(len(wkts), 0)