Skip to content

Commit ba88898

Browse files
committed
add test for defined and reset methods in TensorTest
1 parent ff9c4f2 commit ba88898

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

test/TensorTest.cpp

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,5 +170,20 @@ TEST_F(TensorTest, Transpose) {
170170
EXPECT_EQ(transposed.sizes()[2], 2);
171171
}
172172

173+
// 测试 defined
174+
TEST_F(TensorTest, Defined) {
175+
// Tensor tensor(paddle_tensor_);
176+
177+
EXPECT_TRUE(tensor.defined());
178+
}
179+
180+
// 测试 reset
181+
TEST_F(TensorTest, Reset) {
182+
// Tensor tensor(paddle_tensor_);
183+
184+
tensor.reset();
185+
EXPECT_EQ(tensor.defined(), false);
186+
}
187+
173188
} // namespace test
174189
} // namespace at

0 commit comments

Comments
 (0)