Added 'logging' import to control_flow_ops which is used in the file but not imported.

Change: 110842260

1 files changed, 22 insertions, 0 deletions

diff --git a/tensorflow/python/kernel_tests/reduction_ops_test.py b/tensorflow/python/kernel_tests/reduction_ops_test.py
index 7ff3851da7..3b79ae341b 100644
--- a/tensorflow/python/kernel_tests/reduction_ops_test.py
+++ b/tensorflow/python/kernel_tests/reduction_ops_test.py
@@ -174,6 +174,28 @@ class SumReductionTest(tf.test.TestCase):
   def testGradient4(self):
     self._compareGradient([2, 3, 4, 2], [], None)
 
+  def testHighRank(self):
+    # Do a bunch of random high dimensional reductions
+    np.random.seed(42)
+    for _ in range(20):
+      rank = np.random.randint(4, 10 + 1)
+      axes, = np.nonzero(np.random.randint(2, size=rank))
+      shape = tuple(np.random.randint(1, 3 + 1, size=rank))
+      data = np.random.randint(1024, size=shape)
+      self._compareAll(data, axes)
+    # Check some particular axis patterns
+    for rank in 4, 7, 10:
+      shape = tuple(np.random.randint(1, 3 + 1, size=rank))
+      data = np.random.randint(1024, size=shape)
+      for axes in ([], np.arange(rank), np.arange(0, rank, 2),
+                   np.arange(1, rank, 2)):
+        self._compareAll(data, axes)
+
+  def testExpand(self):
+    # Reduce an empty tensor to a nonempty tensor
+    x = np.zeros((5, 0))
+    self._compareAll(x, [1])
+
 
 class MeanReductionTest(tf.test.TestCase):