diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/padding_fifo_queue_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/padding_fifo_queue_test.py | 124 |
1 files changed, 62 insertions, 62 deletions
diff --git a/tensorflow/python/kernel_tests/padding_fifo_queue_test.py b/tensorflow/python/kernel_tests/padding_fifo_queue_test.py index d8c3f9823c..95f3dcceea 100644 --- a/tensorflow/python/kernel_tests/padding_fifo_queue_test.py +++ b/tensorflow/python/kernel_tests/padding_fifo_queue_test.py @@ -95,13 +95,13 @@ class PaddingFIFOQueueTest(test.TestCase): """, q.queue_ref.op.node_def) def testEnqueue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) enqueue_op = q.enqueue((10.0,)) enqueue_op.run() def testEnqueueWithShape(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue( 10, dtypes_lib.float32, shapes=((3, 2),)) enqueue_correct_op = q.enqueue(([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]],)) @@ -111,14 +111,14 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(1, q.size().eval()) def testEnqueueManyWithShape(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue( 10, [dtypes_lib.int32, dtypes_lib.int32], shapes=[(), (2,)]) q.enqueue_many([[1, 2, 3, 4], [[1, 1], [2, 2], [3, 3], [4, 4]]]).run() self.assertEqual(4, q.size().eval()) def testParallelEnqueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0, 50.0, 60.0, 70.0, 80.0, 90.0, 100.0] enqueue_ops = [q.enqueue((x,)) for x in elems] @@ -144,7 +144,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertItemsEqual(elems, results) def testParallelDequeue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0, 50.0, 60.0, 70.0, 80.0, 90.0, 100.0] enqueue_ops = [q.enqueue((x,)) for x in elems] @@ -168,7 +168,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertItemsEqual(elems, results) def testDequeue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0] enqueue_ops = [q.enqueue((x,)) for x in elems] @@ -182,7 +182,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([elems[i]], vals) def testEnqueueAndBlockingDequeue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(3, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0] enqueue_ops = [q.enqueue((x,)) for x in elems] @@ -212,7 +212,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([elem], result) def testMultiEnqueueAndDequeue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, (dtypes_lib.int32, dtypes_lib.float32), ((), ())) @@ -230,12 +230,12 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([y], y_val) def testQueueSizeEmpty(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) self.assertEqual([0], q.size().eval()) def testQueueSizeAfterEnqueueAndDequeue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) enqueue_op = q.enqueue((10.0,)) dequeued_t = q.dequeue() @@ -248,7 +248,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(0, size.eval()) def testEnqueueMany(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -261,7 +261,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([elems[i % 4]], vals) def testEmptyEnqueueMany(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ( (None, None),)) empty_t = constant_op.constant( @@ -274,7 +274,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([0], size_t.eval()) def testEmptyDequeueMany(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, shapes=((),)) enqueue_op = q.enqueue((10.0,)) dequeued_t = q.dequeue_many(0) @@ -284,7 +284,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([], dequeued_t.eval().tolist()) def testEmptyDequeueManyWithDynamicShape(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue( 10, dtypes_lib.float32, shapes=((None,),)) enqueue_op = q.enqueue(([10.0],)) @@ -295,7 +295,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([], dequeued_t.eval().tolist()) def testEmptyDequeueUpToWithDynamicShape(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue( 10, dtypes_lib.float32, shapes=((None,),)) enqueue_op = q.enqueue(([10.0],)) @@ -306,7 +306,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual([], dequeued_t.eval().tolist()) def testConstructPaddingFIFOQueueWithNoShape(self): - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp( ValueError, r"When providing partial shapes, a list of shapes must be provided."): @@ -314,7 +314,7 @@ class PaddingFIFOQueueTest(test.TestCase): None).queue_ref.eval() def testMultiEnqueueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, (dtypes_lib.float32, dtypes_lib.int32), ((), (2,))) @@ -332,7 +332,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(int_elems[i % 4], int_val) def testMultiEnqueueManyWithPartiallyKnownShapes(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue( 10, (dtypes_lib.float32, dtypes_lib.int32), shapes=((), (None,))) float_elems = [10.0, 20.0, 30.0, 40.0] @@ -349,7 +349,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(int_elems[i % 4], int_val) def testDequeueMany(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0, 50.0, 60.0, 70.0, 80.0, 90.0, 100.0] enqueue_op = q.enqueue_many((elems,)) @@ -361,7 +361,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(elems[4:8], dequeued_t.eval()) def testDequeueUpToNoBlocking(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0, 50.0, 60.0, 70.0, 80.0, 90.0, 100.0] enqueue_op = q.enqueue_many((elems,)) @@ -373,7 +373,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(elems[4:8], dequeued_t.eval()) def testMultiDequeueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue( 10, (dtypes_lib.float32, dtypes_lib.int32), shapes=((), (2,))) float_elems = [ @@ -404,7 +404,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(int_val.shape, dequeued_single_t[1].get_shape()) def testMultiDequeueManyWithPartiallyKnownShapes(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue( 10, (dtypes_lib.float32, dtypes_lib.int32), shapes=((), (None,))) float_elems = [ @@ -443,7 +443,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeued_single_t[1].get_shape())) def testMultiDequeueManyWithPartiallyKnownShapesAndVariableSizeInput(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue( 10, (dtypes_lib.string, dtypes_lib.int32), shapes=((None,), (1, None))) @@ -484,7 +484,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeued_single_t[1].get_shape())) def testMultiDequeueUpToPartiallyKnownShapesAndVariableInputNoBlocking(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue( 10, (dtypes_lib.string, dtypes_lib.int32), shapes=((None,), (1, None))) @@ -525,7 +525,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeued_single_t[1].get_shape())) def testHighDimension(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.int32, ((4, 4, 4, 4),)) elems = np.array([[[[[x] * 4] * 4] * 4] * 4 for x in range(10)], np.int32) enqueue_op = q.enqueue_many((elems,)) @@ -535,7 +535,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(dequeued_t.eval(), elems) def testPartiallyKnownHighDimension(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.int32, ( (4, None, 4, None),)) elems = np.array([[[[[x] * 4] * 4] * 4] * 4 for x in range(10)], np.int32) @@ -592,7 +592,7 @@ class PaddingFIFOQueueTest(test.TestCase): array_ops.placeholder(dtypes_lib.int32))) def testEnqueueWrongPartiallyKnownShapeAtRuntime(self): - with self.test_session() as sess: + with self.cached_session() as sess: # First dimension of second component is unknown, second # dimension must be 3. q = data_flow_ops.PaddingFIFOQueue(10, @@ -607,7 +607,7 @@ class PaddingFIFOQueueTest(test.TestCase): feed_dict={elems_bad: np.array([1] * 12).reshape((3, 4))}) def testEnqueueDequeueManyWrongPartiallyKnownShape(self): - with self.test_session() as sess: + with self.cached_session() as sess: # First dimension of second component is unknown, second # dimension must be 3. q = data_flow_ops.PaddingFIFOQueue(10, @@ -625,7 +625,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeued_t.eval() def testParallelEnqueueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(1000, dtypes_lib.float32, shapes=((),)) elems = [10.0 * x for x in range(100)] enqueue_op = q.enqueue_many((elems,)) @@ -644,7 +644,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertItemsEqual(dequeued_t.eval(), elems * 10) def testParallelDequeueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(1000, dtypes_lib.float32, shapes=((),)) elems = [10.0 * x for x in range(1000)] enqueue_op = q.enqueue_many((elems,)) @@ -666,7 +666,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertItemsEqual(elems, dequeued_elems) def testParallelDequeueUpTo(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(1000, dtypes_lib.float32, shapes=((),)) elems = [10.0 * x for x in range(1000)] enqueue_op = q.enqueue_many((elems,)) @@ -690,7 +690,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertItemsEqual(elems, dequeued_elems) def testParallelEnqueueAndDequeue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(50, dtypes_lib.float32, shapes=((),)) initial_elements = [10.0] * 49 q.enqueue_many((initial_elements,)).run() @@ -723,7 +723,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertTrue(elem in (10.0, 20.0)) def testMixtureOfEnqueueAndEnqueueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.int32, shapes=((),)) enqueue_placeholder = array_ops.placeholder(dtypes_lib.int32, shape=()) enqueue_op = q.enqueue((enqueue_placeholder,)) @@ -759,7 +759,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(0, q.size().eval()) def testMixtureOfDequeueAndDequeueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.int32, shapes=((),)) enqueue_op = q.enqueue_many((np.arange(250, dtype=np.int32),)) dequeued_t = q.dequeue() @@ -793,7 +793,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(0, q.size().eval()) def testBlockingDequeueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -820,7 +820,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(elems, dequeued_elems) def testBlockingDequeueUpTo(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -847,7 +847,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(elems, dequeued_elems) def testDequeueManyWithTensorParameter(self): - with self.test_session(): + with self.cached_session(): # Define a first queue that contains integer counts. dequeue_counts = [random.randint(1, 10) for _ in range(100)] count_q = data_flow_ops.PaddingFIFOQueue(100, dtypes_lib.int32, ((),)) @@ -872,7 +872,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(elems, dequeued_elems) def testDequeueFromClosedQueue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -890,7 +890,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeued_t.eval() def testBlockingDequeueFromClosedQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -916,7 +916,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testDequeueUpToFromClosedQueueReturnsRemainder(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -938,7 +938,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testBlockingDequeueFromClosedEmptyQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) close_op = q.close() dequeued_t = q.dequeue() @@ -958,7 +958,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testBlockingDequeueManyFromClosedQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -983,7 +983,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testBlockingDequeueManyButNotAllFromClosedQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -1008,7 +1008,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testEnqueueManyLargerThanCapacityWithConcurrentDequeueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(4, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -1045,7 +1045,7 @@ class PaddingFIFOQueueTest(test.TestCase): close_thread.join() def testClosedBlockingDequeueManyRestoresPartialBatch(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(4, (dtypes_lib.float32, dtypes_lib.float32), ((), ())) elems_a = [1.0, 2.0, 3.0] @@ -1078,7 +1078,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(0, q.size().eval()) def testBlockingDequeueManyFromClosedEmptyQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) close_op = q.close() dequeued_t = q.dequeue_many(4) @@ -1098,7 +1098,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testBlockingDequeueUpToFromClosedEmptyQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) close_op = q.close() dequeued_t = q.dequeue_up_to(4) @@ -1118,7 +1118,7 @@ class PaddingFIFOQueueTest(test.TestCase): dequeue_thread.join() def testEnqueueToClosedQueue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) enqueue_op = q.enqueue((10.0,)) close_op = q.close() @@ -1131,7 +1131,7 @@ class PaddingFIFOQueueTest(test.TestCase): enqueue_op.run() def testEnqueueManyToClosedQueue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -1145,7 +1145,7 @@ class PaddingFIFOQueueTest(test.TestCase): enqueue_op.run() def testBlockingEnqueueToFullQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(4, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -1168,7 +1168,7 @@ class PaddingFIFOQueueTest(test.TestCase): thread.join() def testBlockingEnqueueManyToFullQueue(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(4, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -1195,7 +1195,7 @@ class PaddingFIFOQueueTest(test.TestCase): thread.join() def testBlockingEnqueueBeforeClose(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(4, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0, 40.0] enqueue_op = q.enqueue_many((elems,)) @@ -1232,7 +1232,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(0, q.size().eval()) def testBlockingEnqueueManyBeforeClose(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(4, dtypes_lib.float32, ((),)) elems = [10.0, 20.0, 30.0] enqueue_op = q.enqueue_many((elems,)) @@ -1265,7 +1265,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(elem, dequeued_t.eval()) def testDoesNotLoseValue(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.PaddingFIFOQueue(1, dtypes_lib.float32, ((),)) enqueue_op = q.enqueue((10.0,)) size_t = q.size() @@ -1275,7 +1275,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(size_t.eval(), [1]) def testSharedQueueSameSession(self): - with self.test_session(): + with self.cached_session(): q1 = data_flow_ops.PaddingFIFOQueue( 1, dtypes_lib.float32, ((),), shared_name="shared_queue") q1.enqueue((10.0,)).run() @@ -1305,7 +1305,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(q2_size_t.eval(), [0]) def testIncompatibleSharedQueueErrors(self): - with self.test_session(): + with self.cached_session(): q_a_1 = data_flow_ops.PaddingFIFOQueue( 10, dtypes_lib.float32, ((),), shared_name="q_a") q_a_2 = data_flow_ops.PaddingFIFOQueue( @@ -1356,7 +1356,7 @@ class PaddingFIFOQueueTest(test.TestCase): q_f_2.queue_ref.op.run() def testSelectQueue(self): - with self.test_session(): + with self.cached_session(): num_queues = 10 qlist = list() for _ in xrange(num_queues): @@ -1370,7 +1370,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertEqual(q.dequeue().eval(), 10.0) def testSelectQueueOutOfRange(self): - with self.test_session(): + with self.cached_session(): q1 = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),)) q2 = data_flow_ops.PaddingFIFOQueue(15, dtypes_lib.float32, ((),)) enq_q = data_flow_ops.PaddingFIFOQueue.from_list(3, [q1, q2]) @@ -1394,7 +1394,7 @@ class PaddingFIFOQueueTest(test.TestCase): sess.run(enqueue_many_op) def testResetOfBlockingOperation(self): - with self.test_session() as sess: + with self.cached_session() as sess: q_empty = data_flow_ops.PaddingFIFOQueue(5, dtypes_lib.float32, ((),)) dequeue_op = q_empty.dequeue() dequeue_many_op = q_empty.dequeue_many(1) @@ -1422,7 +1422,7 @@ class PaddingFIFOQueueTest(test.TestCase): t.join() def testBigEnqueueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(5, dtypes_lib.int32, ((),)) elem = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] enq = q.enqueue_many((elem,)) @@ -1467,7 +1467,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(elem, results) def testBigDequeueMany(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.PaddingFIFOQueue(2, dtypes_lib.int32, ((),)) elem = np.arange(4, dtype=np.int32) enq_list = [q.enqueue((e,)) for e in elem] @@ -1493,7 +1493,7 @@ class PaddingFIFOQueueTest(test.TestCase): self.assertAllEqual(elem, results) def testDtypes(self): - with self.test_session() as sess: + with self.cached_session() as sess: dtypes = [ dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.int32, dtypes_lib.uint8, dtypes_lib.int16, dtypes_lib.int8, dtypes_lib.int64, |