From dd3f71b64072c44cec65a7a883d0c3a29659645c Mon Sep 17 00:00:00 2001
From: Sang-Hoon Park <sang-hoon.park@arm.com>
Date: Tue, 18 Feb 2020 11:27:35 +0000
Subject: COMPMID-3060: Add TF Parser support for Transpose

Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Change-Id: I9661787071554b38c5b0ab3c98431f3863b98520
---
 src/armnnUtils/test/PrototxtConversionsTest.cpp | 97 +++++++++++++++++++++++++
 1 file changed, 97 insertions(+)

(limited to 'src/armnnUtils/test/PrototxtConversionsTest.cpp')

diff --git a/src/armnnUtils/test/PrototxtConversionsTest.cpp b/src/armnnUtils/test/PrototxtConversionsTest.cpp
index e06fbe0f2e..f263a52340 100644
--- a/src/armnnUtils/test/PrototxtConversionsTest.cpp
+++ b/src/armnnUtils/test/PrototxtConversionsTest.cpp
@@ -4,6 +4,7 @@
 //
 
 #include <PrototxtConversions.hpp>
+#include "armnn/Tensor.hpp"
 
 #include <boost/test/unit_test.hpp>
 
@@ -38,4 +39,100 @@ BOOST_AUTO_TEST_CASE(ConvertInt32ToOctalStringTest)
     BOOST_ASSERT(octalString.compare("\\\\000\\\\000\\\\000\\\\377"));
 }
 
+BOOST_AUTO_TEST_CASE(ConvertTensorShapeToStringTest)
+{
+    using armnnUtils::ConvertTensorShapeToString;
+    using armnn::TensorShape;
+
+    auto createAndConvert = [](std::initializer_list<unsigned int> dims) -> std::string
+    {
+        auto shape = TensorShape{dims};
+        return ConvertTensorShapeToString(shape);
+    };
+
+    auto output_string = createAndConvert({5});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+        "size: 5\n"
+        "}"));
+
+    output_string = createAndConvert({4, 5});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+            "size: 4\n"
+        "}\n"
+        "dim {\n"
+            "size: 5\n"
+        "}"
+        ));
+
+    output_string = createAndConvert({3, 4, 5});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+            "size: 3\n"
+        "}\n"
+        "dim {\n"
+            "size: 4\n"
+        "}\n"
+        "dim {\n"
+            "size: 5\n"
+        "}"
+        ));
+
+    output_string = createAndConvert({2, 3, 4, 5});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+            "size: 2\n"
+        "}\n"
+        "dim {\n"
+            "size: 3\n"
+        "}\n"
+        "dim {\n"
+            "size: 4\n"
+        "}\n"
+        "dim {\n"
+            "size: 5\n"
+        "}"
+        ));
+
+    output_string = createAndConvert({1, 2, 3, 4, 5});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+            "size: 1\n"
+        "}\n"
+        "dim {\n"
+            "size: 2\n"
+        "}\n"
+        "dim {\n"
+            "size: 3\n"
+        "}\n"
+        "dim {\n"
+            "size: 4\n"
+        "}\n"
+        "dim {\n"
+            "size: 5\n"
+        "}"
+        ));
+
+    output_string = createAndConvert({0xffffffff, 0xffffffff});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+            "size: 4294967295\n"
+        "}\n"
+        "dim {\n"
+            "size: 4294967295\n"
+        "}"
+        ));
+
+    output_string = createAndConvert({1, 0});
+    BOOST_ASSERT(output_string.compare(
+        "dim {\n"
+            "size: 1\n"
+        "}\n"
+        "dim {\n"
+            "size: 0\n"
+        "}"
+        ));
+}
+
 BOOST_AUTO_TEST_SUITE_END()
-- 
cgit v1.2.1