Skip to content

Commit ed4b247

Browse files
committed
add an unittest
1 parent 8553ac6 commit ed4b247

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

paddle/fluid/platform/enforce_test.cc

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,3 +210,14 @@ TEST(ENFORCE_USER_DEFINED_CLASS, NE) {
210210
Dims a{{1, 2, 3, 4}}, b{{5, 6, 7, 8}};
211211
ASSERT_THROW(PADDLE_ENFORCE_EQ(a, b), paddle::platform::EnforceNotMet);
212212
}
213+
214+
TEST(EOF_EXCEPTION, THROW_EOF) {
215+
bool caught_eof = false;
216+
try {
217+
PADDLE_THROW_EOF();
218+
} catch (paddle::platform::EOFException error) {
219+
caught_eof = true;
220+
EXPECT_TRUE(HasPrefix(StringPiece(error.what()), "There is no next data."));
221+
}
222+
EXPECT_TRUE(caught_eof);
223+
}

0 commit comments

Comments
 (0)