Skip to content

Commit 17c51d6

Browse files
committed
fix unit test error
1 parent 304b6b7 commit 17c51d6

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

paddle/fluid/operators/reader/reader_blocking_queue_test.cc

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,6 @@ TEST(BlockingQueue, CapacityTest) {
2828
EXPECT_EQ(q.Cap(), cap);
2929
}
3030

31-
TEST(BlockingQueue, CanSendTest) {
32-
size_t cap = 1;
33-
BlockingQueue<int> q(cap);
34-
q.Send(1);
35-
EXPECT_FALSE(q.CanSend());
36-
}
37-
3831
void FirstInFirstOut(size_t queue_cap, size_t elem_num, size_t send_time_gap,
3932
size_t receive_time_gap) {
4033
BlockingQueue<size_t> q(queue_cap);
@@ -83,9 +76,11 @@ TEST(BlockingQueue, SenderBlockingTest) {
8376
sender.join();
8477
EXPECT_EQ(send_count, queue_cap);
8578
std::vector<size_t> res;
86-
while (q.CanReceive()) {
79+
while (true) {
8780
size_t elem;
88-
EXPECT_TRUE(q.Receive(&elem));
81+
if (!q.Receive(&elem) {
82+
break;
83+
}
8984
res.push_back(elem);
9085
}
9186
EXPECT_EQ(res.size(), queue_cap);

0 commit comments

Comments
 (0)