[plier-tf] Moving moco test cases regarding data_layout (#6061)

Test cases of moco were moved into plier.
These test cases are for two methods that were moved lately from moco to plier.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>

diff --git a/compiler/plier-tf/src/Convert.test.cpp b/compiler/plier-tf/src/Convert.test.cpp
index c7ca25a..a543db0 100644 (file)
--- a/compiler/plier-tf/src/Convert.test.cpp
+++ b/compiler/plier-tf/src/Convert.test.cpp
@@ -79,6 +79,11 @@ TEST(plier_Convert, attr_shape)
   ASSERT_EQ(shape.dim(3).size(), 8);
 }
 
+TEST(plier_Convert, to_loco_datatype)
+{
+  ASSERT_EQ(plier::tf::as_loco_datatype(tensorflow::DT_FLOAT), loco::DataType::FLOAT32);
+}
+
 TEST(plier_Convert, attr_ilist)
 {
   tensorflow::NodeDef node;
@@ -92,3 +97,9 @@ TEST(plier_Convert, attr_ilist)
   ASSERT_EQ(i_list.at(2), 1LL << 40);
   ASSERT_EQ(i_list.at(3), -(1LL << 40));
 }
+
+TEST(plier_Convert, to_data_layout)
+{
+  ASSERT_EQ(plier::tf::as_data_layout("NHWC"), plier::tf::DataLayout::NHWC);
+  ASSERT_EQ(plier::tf::as_data_layout("NCHW"), plier::tf::DataLayout::NCHW);
+}