Added improved tests

test/tensorflex_test.exs

@@ -1,54 +1,193 @@
 defmodule TensorflexTest do
   use ExUnit.Case
-  doctest Tensorflex
 
-  test "Matrix Functionalities Test" do
-    mat = Tensorflex.create_matrix(2,3, [[2.2,1.3,44.5],[5.5,6.1,3.333]])
-    assert 5.5 = Tensorflex.matrix_pos(mat,2,1)
-    assert {2,3} = Tensorflex.size_of_matrix(mat)
-    assert [[2.2,1.3,44.5],[5.5,6.1,3.333]] = Tensorflex.matrix_to_lists(mat)
+  describe "matrix functionalities" do  
+    test "matrix creation check" do
+      assert [[2.2,1.3,44.5],[5.5,6.1,3.333]] = Tensorflex.create_matrix(2,3, [[2.2,1.3,44.5],[5.5,6.1,3.333]]) |> Tensorflex.matrix_to_lists
+    end
+
+    test "matrix to lists conversion check" do
+      mat = Tensorflex.create_matrix(5,4,[[123,431,23,1],[1,2,3,4],[5,6,7,8],[768,564,44,5],[1,2,3,4]])
+      assert [[123.0, 431.0, 23.0, 1.0],[1.0, 2.0, 3.0, 4.0],[5.0, 6.0, 7.0, 8.0],[768.0, 564.0, 44.0, 5.0],[1.0, 2.0, 3.0, 4.0]] = Tensorflex.matrix_to_lists mat
+    end
+
+    test "matrix access function check" do
+      mat = Tensorflex.create_matrix(2,3, [[2.2,1.3,44.5],[5.5,6.1,3.333]])
+      assert 5.5 = Tensorflex.matrix_pos(mat,2,1)
+      assert 3.333 = Tensorflex.matrix_pos(mat,2,3)
+    end
+
+    test "get size of matrix" do
+      assert {3,3} = Tensorflex.create_matrix(3,3, [[3.9,62,122],[2.2,1.3,44.5],[5.5,6.1,3.333]]) |> Tensorflex.size_of_matrix 
+    end
+
+    test "append new row to matrix function check" do
+      mat = Tensorflex.create_matrix(4,4,[[123,431,23,1],[1,2,3,4],[5,6,7,8],[768,564,44,5]])
+      mat = Tensorflex.append_to_matrix(mat, [[4.4,2,7,9.9]])
+      assert {5,4} = Tensorflex.size_of_matrix mat
+      assert 7.0 = Tensorflex.matrix_pos(mat,5,3)
+    end
   end
 
-  test "Tensor Functionalities Test" do
-    dims = Tensorflex.create_matrix(1,3,[[1,1,3]])
-    vals = Tensorflex.create_matrix(1,3,[[245,202,9]])
-    assert {:ok, :tf_float} = Tensorflex.tensor_datatype(elem(Tensorflex.float32_tensor(vals,dims),1))
-    assert {:ok, :tf_float} = Tensorflex.tensor_datatype(elem(Tensorflex.float32_tensor_alloc(dims),1))
-    assert {:ok, :tf_float} = Tensorflex.tensor_datatype(elem(Tensorflex.float32_tensor(123.123),1))
-    assert {:ok, :tf_string} = Tensorflex.tensor_datatype(elem(Tensorflex.string_tensor("123.123"),1))
-    assert {:ok, :tf_double} = Tensorflex.tensor_datatype(elem(Tensorflex.float64_tensor(123.123),1))
-    assert_raise ArgumentError, fn ->
-      Tensorflex.float32_tensor("123.123")
+  describe "float32 tensor functionalities" do
+    test "float32_tensor/2 tensor creation check" do
+      dims = Tensorflex.create_matrix(1,2,[[1,3]])
+      vals = Tensorflex.create_matrix(1,3,[[245,202,9]])
+      {:ok, tensor} = Tensorflex.float32_tensor vals,dims
+      {:ok, :tf_float} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "float32_tensor/1 tensor creation check" do
+      {:ok, tensor} = Tensorflex.float32_tensor 1234.1234
+      {:ok, :tf_float} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "float32_tensor_alloc/1 tensor creation check" do
+      dims = Tensorflex.create_matrix(1,2,[[1,3]])
+      {:ok, tensor} = Tensorflex.float32_tensor_alloc dims
+      {:ok, :tf_float} = Tensorflex.tensor_datatype(tensor)
     end
-    assert_raise ArgumentError, fn ->
-      Tensorflex.string_tensor(123.123)
+
+    test "incorrect usage check" do
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.float32_tensor("123.123")
+      end
+
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.float32_tensor(123)
+      end
     end
   end
 
-  test "Graph Loading Test" do
-    assert ["biases", "biases/read", "weights", "weights/read", "input", "MatMul", "add", "output"] = Tensorflex.get_graph_ops(elem(Tensorflex.read_graph("./test/graphdef_toy.pb"),1))
-    assert ["biases2", "biases2/read", "weights2", "weights2/read", "biases1", "biases1/read", "weights1", "weights1/read", "input", "MatMul", "Add", "Relu", "MatMul_1", "Add_1", "output"] = Tensorflex.get_graph_ops(elem(Tensorflex.read_graph("./test/graphdef_iris.pb"),1))
-    assert_raise ArgumentError, fn ->
-      Tensorflex.read_graph("Makefile")
+  describe "float64 tensor functionalities" do
+    test "float64_tensor/2 tensor creation check" do
+      dims = Tensorflex.create_matrix(1,2,[[1,3]])
+      vals = Tensorflex.create_matrix(1,3,[[245,202,9]])
+      {:ok, tensor} = Tensorflex.float64_tensor vals,dims
+      {:ok, :tf_double} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "float64_tensor/1 tensor creation check" do
+      {:ok, tensor} = Tensorflex.float64_tensor 1234.1234
+      {:ok, :tf_double} = Tensorflex.tensor_datatype(tensor)
     end
-    assert_raise ArgumentError, fn ->
-      Tensorflex.read_graph("graph.txt")
+
+    test "float64_tensor_alloc/1 tensor creation check" do
+      dims = Tensorflex.create_matrix(1,2,[[1,3]])
+      {:ok, tensor} = Tensorflex.float64_tensor_alloc dims
+      {:ok, :tf_double} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "incorrect usage check" do
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.float64_tensor("123.123")
+      end
+
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.float64_tensor(123)
+      end
     end
   end
 
-  test "Session Test" do
-    {:ok, graph} = Tensorflex.read_graph("./test/graphdef_toy.pb")
-    in_vals = Tensorflex.create_matrix(3,3,[[1.0,1.0,1.0],[2.0,2.0,2.0],[3.0,3.0,3.0]])
-    in_dims = Tensorflex.create_matrix(1,2,[[3,3]])
-    {:ok, input_tensor} = Tensorflex.float32_tensor(in_vals, in_dims)
-    out_dims = Tensorflex.create_matrix(1,2,[[3,2]])
-    {:ok, output_tensor} = Tensorflex.float32_tensor_alloc(out_dims)
-    assert [[56.349998474121094, 39.26000213623047], [109.69999694824219, 75.52000427246094], [163.04998779296875, 111.77999877929688]] = Tensorflex.run_session(graph, input_tensor, output_tensor, "input", "output")
+  describe "int32 tensor functionalities" do
+    test "int32_tensor/2 tensor creation check" do
+      dims = Tensorflex.create_matrix(1,2,[[1,3]])
+      vals = Tensorflex.create_matrix(1,3,[[245,202,9]])
+      {:ok, tensor} = Tensorflex.int32_tensor vals,dims
+      {:ok, :tf_int32} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "int32_tensor/1 tensor creation check" do
+      {:ok, tensor} = Tensorflex.int32_tensor 1234
+      {:ok, :tf_int32} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "int32_tensor_alloc/1 tensor creation check" do
+      dims = Tensorflex.create_matrix(1,2,[[1,3]])
+      {:ok, tensor} = Tensorflex.int32_tensor_alloc dims
+      {:ok, :tf_int32} = Tensorflex.tensor_datatype(tensor)
+    end
+
+    test "incorrect usage check" do
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.int32_tensor("123.123")
+      end
+
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.int32_tensor(123.123)
+      end
+    end
   end
-
-end
 
+  describe "string tensor functionality" do
+    test "string tensor creation check" do
+      {:ok, tensor} = Tensorflex.string_tensor "123.123"
+      {:ok, :tf_string} = Tensorflex.tensor_datatype tensor
+    end
 
+    test "incorrect usage check" do
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.string_tensor(123.123)
+      end
+
+      assert_raise FunctionClauseError, fn ->
+	Tensorflex.string_tensor(123)
+      end
+    end
+  end
 
+  describe "graph loading and reading functionalities" do
+    test "graph loading check" do
+      {:ok, graph_toy} = Tensorflex.read_graph "./test/graphdef_toy.pb"
+      {:ok, graph_iris} = Tensorflex.read_graph "./test/graphdef_iris.pb"
+    end
+
+    test "get all graph ops" do
+      {:ok, graph_toy} = Tensorflex.read_graph "./test/graphdef_toy.pb"
+      {:ok, graph_iris} = Tensorflex.read_graph "./test/graphdef_iris.pb"
+      assert ["input", "weights", "weights/read", "biases", "biases/read", "MatMul", "add", "output"] = Tensorflex.get_graph_ops graph_toy
+      assert ["input", "weights1", "weights1/read", "biases1", "biases1/read", "weights2", "weights2/read", "biases2", "biases2/read", "MatMul", "Add", "Relu", "MatMul_1", "Add_1", "output"] = Tensorflex.get_graph_ops graph_iris
+    end
+
+    test "incorrect usage check" do
+      assert_raise ArgumentError, fn ->
+	{:ok, graph} = Tensorflex.read_graph "Makefile"
+      end
+
+      assert_raise ArgumentError, fn ->
+	{:ok, graph} = Tensorflex.read_graph "Makefile.pb"
+      end
+    end
+  end
+
+  describe "session functionality" do
+    test "running session check" do
+      {:ok, graph} = Tensorflex.read_graph("./test/graphdef_toy.pb")
+      in_vals = Tensorflex.create_matrix(3,3,[[1.0,1.0,1.0],[2.0,2.0,2.0],[3.0,3.0,3.0]])
+      in_dims = Tensorflex.create_matrix(1,2,[[3,3]])
+      {:ok, input_tensor} = Tensorflex.float32_tensor(in_vals, in_dims)
+      out_dims = Tensorflex.create_matrix(1,2,[[3,2]])
+      {:ok, output_tensor} = Tensorflex.float32_tensor_alloc(out_dims)
+      assert [[56.349998474121094, 39.26000213623047], [109.69999694824219, 75.52000427246094], [163.04998779296875, 111.77999877929688]] = Tensorflex.run_session(graph, input_tensor, output_tensor, "input", "output")
+    end
+  end
+
+  describe "miscellaneous functionalities" do
+    test "CSV-with-header loading function check" do
+      m = Tensorflex.load_csv_as_matrix("./test/sample2.csv", header: :true, delimiter: "-")
+      assert [[1.0, 2.0, 3.0, 4.0], [5.0, 6.0, 7.0, 8.0], [9.0, 10.0, 11.0, 12.0]] = Tensorflex.matrix_to_lists m
+    end
+
+    test "CSV-without-header loading function check" do
+      m = Tensorflex.load_csv_as_matrix("./test/sample1.csv", header: :false)
+      assert [[1.0, 2.0, 3.0, 4.0, 5.0],[6.0, 7.0, 8.0, 9.0, 10.0],[11.0, 12.0, 13.0, 14.0, 15.0]] = Tensorflex.matrix_to_lists m
+    end
+
+    test "CSV-to-matrix function incorrect usage check" do
+      assert_raise ArgumentError, fn ->
+	m = Tensorflex.load_csv_as_matrix("./test/sample1.csv", header: :no_header, delimiter: ",")
+      end
+    end
+  end
 
-  
+end