revise preprocess

Author: foldfelis

src/preprocess.jl

@@ -25,9 +25,9 @@ function get_data(; n=1000, Δsamples=2^3, grid_size=div(2^13, Δsamples))
     y_data = collect(read(file, "u")[1:n, 1:Δsamples:end]')
     close(file)
 
-    x_loc_data = Array{Float32, 3}(undef, grid_size, 2, n)
-    x_loc_data[:, 1, :] .= reshape(repeat(LinRange(0, 1, grid_size), n), (grid_size, n))
-    x_loc_data[:, 2, :] .= x_data
+    x_loc_data = Array{Float32, 3}(undef, 2, grid_size, n)
+    x_loc_data[1, :, :] .= reshape(repeat(LinRange(0, 1, grid_size), n), (grid_size, n))
+    x_loc_data[2, :, :] .= x_data
 
     return x_loc_data, y_data
 end

test/preprocess.jl

@@ -1,6 +1,6 @@
 @testset "get data" begin
     xs, ys = get_data()
 
-    @test size(xs) == (1024, 2, 1000)
+    @test size(xs) == (2, 1024, 1000)
     @test size(ys) == (1024, 1000)
 end