Initial version of the Go API. The API is subject to change.

Remaining work to do:
- Generated ops.
- Generated protocol buffers.
- A few calls requiring protocol buffers aren't in this change.
Change: 131066649

1 files changed, 97 insertions, 0 deletions

diff --git a/tensorflow/go/tensor_test.go b/tensorflow/go/tensor_test.go
new file mode 100644
index 0000000000..630d613729
--- /dev/null
+++ b/tensorflow/go/tensor_test.go
@@ -0,0 +1,97 @@
+// Copyright 2016 The TensorFlow Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package tensorflow
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestNewTensor(t *testing.T) {
+	var tests = []struct {
+		shape []int64
+		value interface{}
+	}{
+		{[]int64{}, int8(5)},
+		{[]int64{}, int16(5)},
+		{[]int64{}, int32(5)},
+		{[]int64{}, int64(5)},
+		{[]int64{}, int64(5)},
+		{[]int64{}, uint8(5)},
+		{[]int64{}, uint16(5)},
+		{[]int64{}, float32(5)},
+		{[]int64{}, float64(5)},
+		{[]int64{}, complex(float32(5), float32(6))},
+		{[]int64{}, complex(float64(5), float64(6))},
+		{[]int64{1}, []float64{1}},
+		{[]int64{1}, [1]float64{1}},
+		{[]int64{3, 2}, [][]float64{{1, 2}, {3, 4}, {5, 6}}},
+		{[]int64{2, 3}, [2][3]float64{{1, 2, 3}, {3, 4, 6}}},
+		{[]int64{4, 3, 2}, [][][]float64{
+			{{1, 2}, {3, 4}, {5, 6}},
+			{{7, 8}, {9, 10}, {11, 12}},
+			{{0, -1}, {-2, -3}, {-4, -5}},
+			{{-6, -7}, {-8, -9}, {-10, -11}},
+		}},
+		{[]int64{2, 0}, [][]int64{{}, {}}},
+	}
+
+	var errorTests = []interface{}{
+		struct{ a int }{5},
+		new(int32),
+		new([]int32),
+		// native ints not supported
+		int(5),
+		[]int{5},
+		// uint32 and uint64 are not supported in TensorFlow
+		uint32(5),
+		[]uint32{5},
+		uint64(5),
+		[]uint64{5},
+	}
+
+	for _, test := range tests {
+		tensor, err := NewTensor(test.value)
+		if err != nil {
+			t.Errorf("NewTensor(%v): %v", test.value, err)
+			continue
+		}
+		if !reflect.DeepEqual(test.shape, tensor.Shape()) {
+			t.Errorf("Tensor.Shape(): got %v, want %v", tensor.Shape(), test.shape)
+		}
+
+		// Test that encode and decode gives the same value. We skip arrays because
+		// they're returned as slices.
+		if reflect.TypeOf(test.value).Kind() != reflect.Array {
+			cTensor := tensor.c()
+			gotTensor := newTensorFromC(cTensor)
+			deleteCTensor(cTensor)
+			got := gotTensor.Value()
+			if !reflect.DeepEqual(test.value, got) {
+				t.Errorf("encode/decode: got %v, want %v", got, test.value)
+			}
+		}
+	}
+
+	for _, test := range errorTests {
+		tensor, err := NewTensor(test)
+		if err == nil {
+			t.Errorf("NewTensor(%v): %v", test, err)
+		}
+		if tensor != nil {
+			t.Errorf("NewTensor(%v) = %v, want nil", test, tensor)
+		}
+	}
+}