fixed call to optimize jenk in plotting

wino6687 · Feb 13, 2020 · 1d6c614 · 1d6c614
1 parent 6e04027
commit 1d6c614
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/swepy/classify.py b/swepy/classify.py
@@ -126,7 +126,7 @@ def plot_jenks(image, gvt, interactive=False):
         "yellow",
     ]
     classes_jenk = jenkspy.jenks_breaks(
-        image.ravel(), optimal_jenk(image.ravel(), gvt)
+        image.ravel(), optimal_jenk(image.ravel(), gvt)[0]
     )
     classes = np.digitize(image, classes_jenk)
     nclasses = len(classes_jenk)

diff --git a/swepy/tests/test_classify.py b/swepy/tests/test_classify.py
@@ -7,7 +7,6 @@
 
 @pytest.fixture
 def swe():
-    print(os.listdir())
     swe = np.load("swepy/tests/swe_testfile.npy")
     return swe
 
@@ -37,11 +36,11 @@ def test_optimal_jenk_bounds(swe):
     """
     info = classify.optimal_jenk(swe.ravel(), 0.8)
     assert info[1] == [
-        22.024993896484375,
-        27.162506103515625,
-        30.257492065429688,
-        33.41499328613281,
-        38.93499755859375,
+        13.257644653320312,
+        16.314910888671875,
+        18.693313598632812,
+        21.749465942382812,
+        28.034332275390625,
     ]