diff --git a/test/TestOpTensorCreate.cpp b/test/TestOpTensorCreate.cpp index a57142c..c1dd200 100644 --- a/test/TestOpTensorCreate.cpp +++ b/test/TestOpTensorCreate.cpp @@ -5,20 +5,19 @@ TEST(TestOpTensorCreate, CreateSingleTensorSingleOp) { - - kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; - std::shared_ptr tensorA{ new kp::Tensor(testVecA) }; - mgr.rebuildTensors({ tensorA }); + { + kp::Manager mgr; - EXPECT_TRUE(tensorA->isInit()); + mgr.rebuildTensors({ tensorA }); - EXPECT_EQ(tensorA->data(), testVecA); + EXPECT_TRUE(tensorA->isInit()); + + EXPECT_EQ(tensorA->data(), testVecA); + } - tensorA->freeMemoryDestroyGPUResources(); EXPECT_FALSE(tensorA->isInit()); }