fix HeatMap constructor for iterator

Author: joshday

Project.toml

@@ -1,6 +1,6 @@
 name = "OnlineStats"
 uuid = "a15396b6-48d5-5d58-9928-6d29437db91e"
-version = "1.5.11"
+version = "1.5.12"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"

src/viz/heatmap.jl

@@ -37,8 +37,8 @@ end
 _heatmap_range(x, bins) = range(extrema(x)..., length=bins)
 
 function HeatMap(itr, bins::Integer=200; kw...)
-    xrng = _heatmap_range(first.(itr))
-    yrng = _heatmap_range(last.(itr))
+    xrng = _heatmap_range(first.(itr), bins)
+    yrng = _heatmap_range(last.(itr), bins)
     fit!(HeatMap(xrng, yrng; kw...), itr)
 end
 nobs(o::HeatMap) = sum(o.counts) + o.out
@@ -107,4 +107,4 @@ end
             o.yedges, vec(sum(o.counts, dims=1))
         end
     end
-end
+end

test/runtests.jl

@@ -178,6 +178,7 @@ end
     data1 = zip(ymat[:,1], ymat[:,2])
     data2 = zip(ymat2[:,1], ymat2[:,2])
     @test ==(mergevals(HeatMap(-5:.1:5, -5:.1:5), data1, data2)...)
+    @test nobs(HeatMap(data1)) == length(data1)
 end
 #-----------------------------------------------------------------------# Hist
 @testset "Hist" begin