Add import for tests and expand cli test

Vajuw · Mar 21, 2017 · b9020cf · b9020cf
1 parent 5ffce1a
commit b9020cf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -34,14 +34,15 @@ def test_cli_feature_stdin():
     warnings.simplefilter('ignore')
     result = runner.invoke(zonalstats,
                            ['--raster', raster,
-                            '--stats', 'mean',
+                            '--stats', 'all',
                             '--prefix', 'test_'],
                            input=open(vector, 'r').read())
     assert result.exit_code == 0
     outdata = json.loads(result.output)
     assert len(outdata['features']) == 1
     feature = outdata['features'][0]
     assert 'test_mean' in feature['properties']
+    assert 'test_std' in feature['properties']
 
 
 def test_cli_features_sequence():

diff --git a/tests/test_zonal.py b/tests/test_zonal.py
@@ -10,6 +10,7 @@
 from rasterstats.utils import VALID_STATS
 from rasterstats.io import read_featurecollection, read_features
 from shapely.geometry import Polygon
+from affine import Affine
 
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))