def testStructuredTensorLikeStructuredTensor(self): """Fully test structured_tensor_array_like.""" foo = structured_tensor.StructuredTensor.from_pyval([{ "a": 3 }, { "a": 7 }]) result = structured_array_ops._structured_tensor_like(foo) self.assertAllEqual([{}, {}], result)
def testStructuredTensorArrayLike(self): """There was a bug in a case in a private function. This was difficult to reach externally, so I wrote a test to check it directly. """ rt = ragged_tensor.RaggedTensor.from_row_splits( array_ops.zeros(shape=[5, 3]), [0, 3, 5]) result = structured_array_ops._structured_tensor_like(rt) self.assertEqual(3, result.rank)
def testStructuredTensorArrayRankZero(self): """Fully test structured_tensor_array_like.""" foo = array_ops.zeros(shape=[]) result = structured_array_ops._structured_tensor_like(foo) self.assertAllEqual({}, result)
def my_fun(my_shape): my_zeros = array_ops.zeros(my_shape) return structured_array_ops._structured_tensor_like(my_zeros)
def testStructuredTensorArrayRankOneKnownShape(self): """Fully test structured_tensor_array_like.""" foo = array_ops.zeros(shape=[4]) result = structured_array_ops._structured_tensor_like(foo) self.assertAllEqual([{}, {}, {}, {}], result)
def my_fun(foo): bar_shape = math_ops.range(foo) bar = array_ops.zeros(shape=bar_shape) structured_array_ops._structured_tensor_like(bar)