no pdb, test for tifs

perrygeo · perrygeo · commit 8018a7c77def · 2015-10-15T16:09:48.000-04:00
diff --git a/setup.cfg b/setup.cfg
@@ -1,4 +1,4 @@
 [pytest]
 norecursedirs = examples* src* scripts* docs*
-addopts = --maxfail=1 --pdb -rf -v
+# addopts = --maxfail=1 --pdb -rf -v
 
diff --git a/src/pyimpute/_main.py b/src/pyimpute/_main.py
@@ -186,7 +186,7 @@ def impute(target_xs, clf, raster_info, outdir="output", linechunk=1000, class_p
             classes = list(clf.classes_)
             class_paths = []
             for i, c in enumerate(classes):
-                ods = os.path.join(outdir, "probability_%s.img" % c)
+                ods = os.path.join(outdir, "probability_%s.tif" % c)
                 class_paths.append(ods)
             for p in class_paths:
                 class_dss.append(rasterio.open(p, 'w', **profile))
diff --git a/tests/test_impute.py b/tests/test_impute.py
@@ -27,9 +27,9 @@ def test_impute():
     impute(target_xs, clf, raster_info, outdir=TMPOUT,
            linechunk=400, class_prob=True, certainty=True)
 
-    assert os.path.exists(os.path.join(TMPOUT, "responses.img"))
-    assert os.path.exists(os.path.join(TMPOUT, "certainty.img"))
-    assert os.path.exists(os.path.join(TMPOUT, "probability_90.img"))
+    assert os.path.exists(os.path.join(TMPOUT, "responses.tif"))
+    assert os.path.exists(os.path.join(TMPOUT, "certainty.tif"))
+    assert os.path.exists(os.path.join(TMPOUT, "probability_90.tif"))
 
 
 def test_load_training_rasters():
