adding unique method to raster class

wbierbower · Apr 30, 2015 · 2852d52 · 2852d52
1 parent 19b38e8
commit 2852d52
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/fauxgeo/raster.py b/fauxgeo/raster.py
@@ -381,6 +381,10 @@ def std(self):
         _, _, _, std = pygeo.get_statistics_from_uri(self.uri)
         return std
 
+    def unique(self):
+        unique_vals_list = pygeo.unique_raster_values_uri(self.uri)
+        return unique_vals_list
+
     def ones(self):
         def ones_closure(nodata):
             def ones(x):

diff --git a/tests/test_raster.py b/tests/test_raster.py
@@ -402,6 +402,7 @@ def test_sum(self):
         print raster
         assert(raster.sum() == 8.0)
 
+
 class TestRasterMinimum(unittest.TestCase):
     def setUp(self):
         self.shape = (4, 4)
@@ -428,5 +429,21 @@ def test_fminimum(self):
         assert(min_raster.get_band(1).data[0, 1] == 2)
 
 
+class TestRasterUnique(unittest.TestCase):
+    def setUp(self):
+        self.shape = (4, 4)
+        self.array = np.ones(self.shape)
+        self.affine = Affine(1, 0, 0, 0, -1, 4)
+        self.proj = 4326
+        self.datatype = gdal.GDT_Int16
+        self.nodata_val = -9999
+        self.factory = RasterFactory(
+            self.proj, self.datatype, self.nodata_val, *self.shape, affine=self.affine)
+
+    def test_unique(self):
+        raster = self.factory.horizontal_ramp(1, 4)
+        assert(raster.unique() == [1, 2, 3, 4])
+
+
 if __name__ == '__main__':
     unittest.main()