diff options
Diffstat (limited to 'tensorflow/contrib/slim/python/slim/data/parallel_reader_test.py')
| -rw-r--r-- | tensorflow/contrib/slim/python/slim/data/parallel_reader_test.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/tensorflow/contrib/slim/python/slim/data/parallel_reader_test.py b/tensorflow/contrib/slim/python/slim/data/parallel_reader_test.py index 10ea883e1f..ea8cc0ff61 100644 --- a/tensorflow/contrib/slim/python/slim/data/parallel_reader_test.py +++ b/tensorflow/contrib/slim/python/slim/data/parallel_reader_test.py @@ -119,9 +119,13 @@ class ParallelReaderTest(test.TestCase): self.assertEquals(count0, num_records_per_file) self.assertEquals(count1, num_records_per_file) self.assertEquals(count2, num_records_per_file) - self.assertEquals(all_keys_count, num_files * num_records_per_file) + self.assertEquals( + all_keys_count, + num_files * num_records_per_file) self.assertEquals(all_values_count, all_keys_count) - self.assertEquals(count0 + count1 + count2, all_keys_count) + self.assertEquals( + count0 + count1 + count2, + all_keys_count) def testRandomShuffleQueue(self): shared_queue = data_flow_ops.RandomShuffleQueue( @@ -140,16 +144,14 @@ class ParallelReaderTest(test.TestCase): capacity=55, min_after_dequeue=28, dtypes=[dtypes_lib.string, dtypes_lib.string], - shapes=[tensor_shape.scalar(), - tensor_shape.scalar()]) + shapes=[tensor_shape.scalar(), tensor_shape.scalar()]) self._verify_read_up_to_out(shared_queue) def testReadUpToFromFIFOQueue(self): shared_queue = data_flow_ops.FIFOQueue( capacity=99, dtypes=[dtypes_lib.string, dtypes_lib.string], - shapes=[tensor_shape.scalar(), - tensor_shape.scalar()]) + shapes=[tensor_shape.scalar(), tensor_shape.scalar()]) self._verify_read_up_to_out(shared_queue) |