use pairs in findmin and findmax, supporting all indexable coll…

…ections
JuliaLang · Jul 21, 2017 · 51be6ac · 51be6ac
1 parent 349592f
commit 51be6ac
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/base/array.jl b/base/array.jl
@@ -1717,12 +1717,12 @@ function findmax(a)
     if isempty(a)
         throw(ArgumentError("collection must be non-empty"))
     end
-    s = start(a)
-    mi = i = 1
-    m, s = next(a, s)
-    while !done(a, s)
-        ai, s = next(a, s)
-        i += 1
+    p = pairs(a)
+    s = start(p)
+    (mi, m), s = next(p, s)
+    i = mi
+    while !done(p, s)
+        (i, ai), s = next(p, s)
         if ai > m || m!=m
             m = ai
             mi = i
@@ -1756,12 +1756,12 @@ function findmin(a)
     if isempty(a)
         throw(ArgumentError("collection must be non-empty"))
     end
-    s = start(a)
-    mi = i = 1
-    m, s = next(a, s)
-    while !done(a, s)
-        ai, s = next(a, s)
-        i += 1
+    p = pairs(a)
+    s = start(p)
+    (mi, m), s = next(p, s)
+    i = mi
+    while !done(p, s)
+        (i, ai), s = next(p, s)
         if ai < m || m!=m
             m = ai
             mi = i

diff --git a/test/arrayops.jl b/test/arrayops.jl
@@ -1796,6 +1796,11 @@ s, si = findmax(S)
 @test a == b == s
 @test ai == bi == si
 
+for X in (A, B, S)
+    @test findmin(X) == findmin(Dict(pairs(X)))
+    @test findmax(X) == findmax(Dict(pairs(X)))
+end
+
 fill!(B, 2)
 @test all(x->x==2, B)