tests: test equal thresholds in discrete transform

GeoStat-Framework · Feb 25, 2020 · ac8fe5c · ac8fe5c
1 parent 2f08ebf
commit ac8fe5c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_srf.py b/tests/test_srf.py
@@ -267,9 +267,8 @@ def test_transform(self):
 
         srf((self.x_grid, self.y_grid), seed=self.seed, mesh_type="structured")
         values = [-1, 0, 0.5, 1]
-        self.assertRaises(
-            NotImplementedError, tf.discrete, srf, values, thresholds="equal"
-        )
+        tf.discrete(srf, values, thresholds="equal")
+        np.testing.assert_array_equal(np.unique(srf.field), values)
 
     def test_incomprrandmeth(self):
         self.cov_model = Gaussian(dim=2, var=0.5, len_scale=1.0)