diff options
author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2015-09-23 13:40:16 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2015-09-23 20:24:10 -0700 |
commit | ae77b15495d4c2a83202c49991bfc0885765de03 (patch) | |
tree | c5aa6d0811406117e5afb01bc810f2695549f86f | |
parent | 74e174537418b6a3c0c8708e444edf45ab491e94 (diff) | |
download | caffeonacl-ae77b15495d4c2a83202c49991bfc0885765de03.tar.gz caffeonacl-ae77b15495d4c2a83202c49991bfc0885765de03.tar.bz2 caffeonacl-ae77b15495d4c2a83202c49991bfc0885765de03.zip |
[test] TestReshape: expect instead of check
-rw-r--r-- | src/caffe/test/test_net.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/test/test_net.cpp b/src/caffe/test/test_net.cpp index ec01053c..16c1d35f 100644 --- a/src/caffe/test/test_net.cpp +++ b/src/caffe/test/test_net.cpp @@ -2306,7 +2306,7 @@ TYPED_TEST(NetTest, TestReshape) { this->net_->ForwardPrefilled(); this->net_->Backward(); for (int i = 0; i < output1.count(); ++i) { - CHECK_EQ(*(output1.cpu_data() + i), *(output_blob->cpu_data() + i)); + EXPECT_FLOAT_EQ(*(output1.cpu_data() + i), *(output_blob->cpu_data() + i)); } input_blob->Reshape(blob2.num(), blob2.channels(), blob2.height(), @@ -2315,7 +2315,7 @@ TYPED_TEST(NetTest, TestReshape) { this->net_->ForwardPrefilled(); this->net_->Backward(); for (int i = 0; i < output2.count(); ++i) { - CHECK_EQ(*(output2.cpu_data() + i), *(output_blob->cpu_data() + i)); + EXPECT_FLOAT_EQ(*(output2.cpu_data() + i), *(output_blob->cpu_data() + i)); } } |