Skip to content

Commit c61e16b

Browse files
committed
add reader_queue_speed_test_mode_flag test
1 parent 8686f7c commit c61e16b

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

paddle/fluid/operators/reader/reader_blocking_queue_test.cc

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -221,3 +221,28 @@ TEST(BlockingQueue, MyClassTest) {
221221
q.Receive(&b);
222222
EXPECT_EQ(a.val_, b.val_);
223223
}
224+
225+
TEST(BlockingQueue, reader_queue_speed_test_mode_flag) {
226+
FLAGS_reader_queue_speed_test_mode = false;
227+
size_t queue_size = 10;
228+
BlockingQueue<size_t> q(queue_size);
229+
for (size_t i = 0; i < queue_size; ++i) {
230+
q.Send(i);
231+
}
232+
size_t b;
233+
for (size_t i = 0; i < queue_size; ++i) {
234+
q.Receive(&b);
235+
EXPECT_EQ(b, i);
236+
}
237+
EXPECT_EQ(q.Size(), 0);
238+
239+
FLAGS_reader_queue_speed_test_mode = true;
240+
for (size_t i = 0; i < queue_size; ++i) {
241+
q.Send(i);
242+
}
243+
for (size_t i = 0; i < queue_size; ++i) {
244+
q.Receive(&b);
245+
EXPECT_EQ(b, 0);
246+
}
247+
EXPECT_EQ(q.Size(), queue_size);
248+
}

0 commit comments

Comments
 (0)