diff --git a/paddle/fluid/operators/reader/reader_blocking_queue_test.cc b/paddle/fluid/operators/reader/reader_blocking_queue_test.cc index 07c22fe2bfb26f76012fe7d3b80a15d72f8c84b7..c0229f53f25e3a2d036eab3743aa71839a6fa7e4 100644 --- a/paddle/fluid/operators/reader/reader_blocking_queue_test.cc +++ b/paddle/fluid/operators/reader/reader_blocking_queue_test.cc @@ -28,13 +28,6 @@ TEST(BlockingQueue, CapacityTest) { EXPECT_EQ(q.Cap(), cap); } -TEST(BlockingQueue, CanSendTest) { - size_t cap = 1; - BlockingQueue q(cap); - q.Send(1); - EXPECT_FALSE(q.CanSend()); -} - void FirstInFirstOut(size_t queue_cap, size_t elem_num, size_t send_time_gap, size_t receive_time_gap) { BlockingQueue q(queue_cap); @@ -83,9 +76,11 @@ TEST(BlockingQueue, SenderBlockingTest) { sender.join(); EXPECT_EQ(send_count, queue_cap); std::vector res; - while (q.CanReceive()) { + while (true) { size_t elem; - EXPECT_TRUE(q.Receive(&elem)); + if (!q.Receive(&elem) { + break; + } res.push_back(elem); } EXPECT_EQ(res.size(), queue_cap);