diff --git a/src/caffeine/test_blob.cpp b/src/caffeine/test_blob.cpp index cd1333b4..38600317 100644 --- a/src/caffeine/test_blob.cpp +++ b/src/caffeine/test_blob.cpp @@ -36,6 +36,14 @@ TYPED_TEST(BlobSimpleTest, TestInitialization) { EXPECT_EQ(this->blob_->count(), 0); } +TYPED_TEST(BlobSimpleTest, TestPointers) { + EXPECT_TRUE(this->blob_preshaped_->gpu_data() != NULL); + EXPECT_TRUE(this->blob_preshaped_->cpu_data() != NULL); + EXPECT_TRUE(this->blob_preshaped_->mutable_gpu_data() != NULL); + EXPECT_TRUE(this->blob_preshaped_->mutable_cpu_data() != NULL); +} + + TYPED_TEST(BlobSimpleTest, TestReshape) { this->blob_->Reshape(2, 3, 4, 5); EXPECT_EQ(this->blob_->num(), 2); @@ -45,4 +53,5 @@ TYPED_TEST(BlobSimpleTest, TestReshape) { EXPECT_EQ(this->blob_->count(), 120); } + }