aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/periodic_resample
diff options
context:
space:
mode:
authorGravatar Jianwei Xie <xiejw@google.com>2018-01-24 11:31:06 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-24 11:35:51 -0800
commitad07a86d75ab06bbcfd6f8f6a24debd9036a52d0 (patch)
tree87041ba0e9c329c3838664345ecc7b5ea1b28787 /tensorflow/contrib/periodic_resample
parent7f8e600bfb4ff5973bd1ec178b65538e2446fb69 (diff)
Fixed linter errors.
PiperOrigin-RevId: 183115307
Diffstat (limited to 'tensorflow/contrib/periodic_resample')
-rw-r--r--tensorflow/contrib/periodic_resample/python/kernel_tests/periodic_resample_op_test.py39
1 files changed, 16 insertions, 23 deletions
diff --git a/tensorflow/contrib/periodic_resample/python/kernel_tests/periodic_resample_op_test.py b/tensorflow/contrib/periodic_resample/python/kernel_tests/periodic_resample_op_test.py
index 30a2077570..a25de55e18 100644
--- a/tensorflow/contrib/periodic_resample/python/kernel_tests/periodic_resample_op_test.py
+++ b/tensorflow/contrib/periodic_resample/python/kernel_tests/periodic_resample_op_test.py
@@ -53,12 +53,11 @@ class PeriodicResampleTest(test_util.TensorFlowTestCase):
def testPeriodicResampleBasic3D(self):
- input_tensor = numpy.arange(2*2*4).reshape((2, 2, 4))
+ input_tensor = numpy.arange(2 * 2 * 4).reshape((2, 2, 4))
desired_shape = numpy.array([4, 4, None])
- output_tensor = numpy.array([[[0], [2], [4], [6]],
- [[1], [3], [5], [7]],
- [[8], [10], [12], [14]],
- [[9], [11], [13], [15]]])
+ output_tensor = numpy.array([[[0], [2], [4], [6]], [[1], [3], [5], [7]],
+ [[8], [10], [12], [14]], [[9], [11], [13],
+ [15]]])
# NOTE: output_tensor != input_tensor.reshape((4, 4, -1))
with self.test_session():
@@ -72,24 +71,18 @@ class PeriodicResampleTest(test_util.TensorFlowTestCase):
def testPeriodicResampleBasic4D(self):
- input_tensor = numpy.arange(2*2*2*8).reshape((2, 2, 2, 8))
+ input_tensor = numpy.arange(2 * 2 * 2 * 8).reshape((2, 2, 2, 8))
desired_shape = numpy.array([4, 4, 4, None])
- output_tensor = numpy.array([[[[0], [4], [8], [12]],
- [[2], [6], [10], [14]],
- [[16], [20], [24], [28]],
- [[18], [22], [26], [30]]],
- [[[1], [5], [9], [13]],
- [[3], [7], [11], [15]],
- [[17], [21], [25], [29]],
- [[19], [23], [27], [31]]],
- [[[32], [36], [40], [44]],
- [[34], [38], [42], [46]],
- [[48], [52], [56], [60]],
- [[50], [54], [58], [62]]],
- [[[33], [37], [41], [45]],
- [[35], [39], [43], [47]],
- [[49], [53], [57], [61]],
- [[51], [55], [59], [63]]]])
+ output_tensor = numpy.array(
+ [[[[0], [4], [8], [12]], [[2], [6], [10], [14]],
+ [[16], [20], [24], [28]], [[18], [22], [26], [30]]],
+ [[[1], [5], [9], [13]], [[3], [7], [11], [15]], [[17], [21], [25],
+ [29]],
+ [[19], [23], [27],
+ [31]]], [[[32], [36], [40], [44]], [[34], [38], [42], [46]],
+ [[48], [52], [56], [60]], [[50], [54], [58], [62]]],
+ [[[33], [37], [41], [45]], [[35], [39], [43], [47]],
+ [[49], [53], [57], [61]], [[51], [55], [59], [63]]]])
# NOTE: output_tensor != input_tensor.reshape((4, 4, 4, -1))
with self.test_session():
@@ -111,5 +104,5 @@ class PeriodicResampleTest(test_util.TensorFlowTestCase):
periodic_resample(input_tensor, [None, 4, 4]).eval()
-if __name__ == "__main__":
+if __name__ == '__main__':
googletest.main()