plot utilities test

src/BiomolecularStructures.jl

@@ -2,4 +2,5 @@ module BiomolecularStructures
 	# include sub-modules
 	include(Pkg.dir("BiomolecularStructures", "src/KABSCH/", "kabsch.jl"))	
 	include(Pkg.dir("BiomolecularStructures", "src/PDB/", "pdb.jl"))	
+	include(Pkg.dir("BiomolecularStructures", "src/PLOT/", "plot.jl"))	
 end

src/PLOT/plot.jl

@@ -1,6 +1,6 @@
 module MatrixPlot
 using PyPlot
-export matrices_plot
+export matrices_plot, matrix_prep
 
 # convert a matrix to PyPlot format
 function matrix_prep(A::Array{Float64,2})
@@ -22,7 +22,7 @@ function matrices_plot(P::Array{Float64,2},Q::Array{Float64,2})
 	pygui(true)
 	x::Array{Float64,1}, y::Array{Float64,1}, z::Array{Float64,1} = matrix_prep(P)
 	scatter3D(x, y, z, color="red")
-	x::Array{Float64,1}, y::Array{Float64,1}, z::Array{Float64,1} = matrix_prep(Q)
+	x,y,z = matrix_prep(Q)
 	scatter3D(x, y, z, color="green")
 	readline()
 end

test/plot.jl

@@ -0,0 +1,9 @@
+using BiomolecularStructures.MatrixPlot
+
+A = [1.0 1.0 1.0; 1.0 1.0 1.0; 1.0 1.0 1.0]
+
+x, y, z = matrix_prep(A)
+
+@test x == [1.0, 1.0, 1.0]
+@test y == [1.0, 1.0, 1.0]
+@test z == [1.0, 1.0, 1.0]

test/runtests.jl

@@ -1,3 +1,4 @@
 using Base.Test
 
-include(Pkg.dir("BiomolecularStructures", "test", "kabsch.jl"))
+include(Pkg.dir("BiomolecularStructures", "test", "kabsch.jl"))
+include(Pkg.dir("BiomolecularStructures", "test", "plot.jl"))