@@ -71,7 +71,7 @@ def test_field_listing():
7171def test_total_feature_count_shp ():
7272 if 'shape' in mapnik .DatasourceCache .plugin_names ():
7373 ds = mapnik .Shapefile (file = '../data/shp/poly.shp' )
74- features = ds . all_features ( )
74+ features = iter ( ds )
7575 num_feats = len (list (features ))
7676 assert num_feats == 10
7777
@@ -83,7 +83,7 @@ def test_total_feature_count_json():
8383 assert desc ['name' ] == 'ogr'
8484 assert desc ['type' ] == mapnik .DataType .Vector
8585 assert desc ['encoding' ] == 'utf-8'
86- features = ds . all_features ( )
86+ features = iter ( ds )
8787 num_feats = len (list (features ))
8888 assert num_feats == 5
8989
@@ -98,7 +98,7 @@ def test_sqlite_reading():
9898 assert desc ['name' ] == 'sqlite'
9999 assert desc ['type' ] == mapnik .DataType .Vector
100100 assert desc ['encoding' ] == 'utf-8'
101- features = ds . all_features ( )
101+ features = iter ( ds )
102102 num_feats = len (list (features ))
103103 assert num_feats == 245
104104
@@ -108,16 +108,15 @@ def test_reading_json_from_string():
108108 json = f .read ()
109109 if 'ogr' in mapnik .DatasourceCache .plugin_names ():
110110 ds = mapnik .Ogr (file = json , layer_by_index = 0 )
111- features = ds . all_features ( )
111+ features = iter ( ds )
112112 num_feats = len (list (features ))
113113 assert num_feats == 5
114114
115115
116116def test_feature_envelope ():
117117 if 'shape' in mapnik .DatasourceCache .plugin_names ():
118118 ds = mapnik .Shapefile (file = '../data/shp/poly.shp' )
119- features = ds .all_features ()
120- for feat in features :
119+ for feat in ds :
121120 env = feat .envelope ()
122121 contains = ds .envelope ().contains (env )
123122 assert contains == True
@@ -128,19 +127,19 @@ def test_feature_envelope():
128127def test_feature_attributes ():
129128 if 'shape' in mapnik .DatasourceCache .plugin_names ():
130129 ds = mapnik .Shapefile (file = '../data/shp/poly.shp' )
131- features = list (ds . all_features ( ))
130+ features = list (iter ( ds ))
132131 feat = features [0 ]
133- attrs = {'PRFEDEA ' : u'35043411' , 'EAS_ID' : 168 , 'AREA ' : 215229.266 }
132+ attrs = {'AREA ' : 215229.266 , 'EAS_ID' : 168 , 'PRFEDEA ' : '35043411' }
134133 assert feat .attributes == attrs
135- assert ds .fields (), ['AREA' , 'EAS_ID' == 'PRFEDEA' ]
136- assert ds .field_types (), ['float' , 'int' == 'str' ]
134+ assert ds .fields (), ['AREA' , 'EAS_ID' , 'PRFEDEA' ]
135+ assert ds .field_types (), ['float' , 'int' , 'str' ]
137136
138137
139138def test_ogr_layer_by_sql ():
140139 if 'ogr' in mapnik .DatasourceCache .plugin_names ():
141140 ds = mapnik .Ogr (file = '../data/shp/poly.shp' ,
142141 layer_by_sql = 'SELECT * FROM poly WHERE EAS_ID = 168' )
143- features = ds . all_features ( )
142+ features = iter ( ds )
144143 num_feats = len (list (features ))
145144 assert num_feats == 1
146145
0 commit comments