Fix Issue #2

haberdashPI · Jul 23, 2019 · 23ef6ad · 23ef6ad
1 parent 8227033
commit 23ef6ad
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/PlotAxes.jl b/src/PlotAxes.jl
@@ -228,7 +228,7 @@ function asplotable(x::AxisArray,ax1,axes...;
   vals = PlotAxes.quantize(x,steps)
   axqvals = PlotAxes.quantize.(axisvalues(x),steps)
 
-  df = DataFrame(value = vec(vals))
+  df = DataFrame(value = convert(Array{eltype(vals)},vec(vals)))
   for ax in show_axes
     axi = findfirst(isequal(axarg_name(ax)),axisnames(x))
     df[!,ax] .= default_value(eltype(axqvals[axi]))

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -84,6 +84,11 @@ end
   data = AxisArray(rand(10),Axis{:time}(range(0u"s",1u"s",length=10)))
   df, = PlotAxes.asplotable(data,quantize=(5,))
   @test size(df,1) == 5
+  @test df.time isa Array
+
+  df, = PlotAxes.asplotable(data,quantize=(20,))
+  @test size(df,1) == length(data)
+  @test df.time isa Array
 
   df, = PlotAxes.asplotable(rand(10,10),quantize=(5,5))
   @test size(df,1) == 25