[coco] KernelObjectTest with comparing Shape (#2952)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Wed, 30 Jan 2019 11:18:47 +0000 (20:18 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Wed, 30 Jan 2019 11:18:47 +0000 (20:18 +0900)
This commit applies TODO comment. KernelObjectTest now compares Shape
with operator==.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
contrib/coco/core/src/IR/KernelObject.test.cpp

index 4aeba23..d1fa24e 100644 (file)
@@ -52,12 +52,7 @@ TEST_F(KernelObjectTest, constructor)
   const nncc::core::ADT::kernel::Shape shape{1, 1, 3, 3};
   auto o = allocate(shape);
 
-  // TODO Use ASSERT_EQ(o.shape(), shape) (operator== overload is not yet available)
-  ASSERT_EQ(o->shape().depth(), shape.depth());
-  ASSERT_EQ(o->shape().count(), shape.count());
-  ASSERT_EQ(o->shape().height(), shape.height());
-  ASSERT_EQ(o->shape().width(), shape.width());
-
+  ASSERT_EQ(o->shape(), shape);
   ASSERT_EQ(o->kind(), coco::Object::Kind::Kernel);
 }