fixed test

JuliaIO · May 22, 2020 · 1930cc7 · 1930cc7
1 parent bcd9a5c
commit 1930cc7
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/test/test_writer.jl b/test/test_writer.jl
@@ -33,20 +33,31 @@ function test_write()
     # the file is very small so only one rowgroup
     col_chunks = columns(pf, 1)
 
+    colnum=12
+    col_chunk=col_chunks[colnum]
+
+    correct_vals = tbl[colnum]
+    coltype = eltype(correct_vals)
+    vals_from_file = values(pf, col_chunk)
+
+    if Missing <: coltype
+        @test ismissing.(correct_vals) == (vals_from_file[2] .== 0)
+    end
 
     for (colnum, col_chunk) in enumerate(col_chunks)
-        println(colnum)
         correct_vals = tbl[colnum]
         coltype = eltype(correct_vals)
         vals_from_file = values(pf, col_chunk)
         if Missing <: coltype
             @test ismissing.(correct_vals) == (vals_from_file[2] .== 0)
         end
 
+        non_missing_vals = collect(skipmissing(correct_vals))
+
         if nonmissingtype(coltype) == String
-            @test all(skipmissing(correct_vals) .== String.(vals_from_file[1]))
+            non_missing_vals_read = String.(vals_from_file[1][1:sum(vals_from_file[2])])
+            @test all(non_missing_vals .== non_missing_vals_read)
         else
-            non_missing_vals = collect(skipmissing(correct_vals))
             non_missing_vals_read = vals_from_file[1][1:sum(vals_from_file[2])]
             @test all(non_missing_vals .== non_missing_vals_read)
         end