PrototxtConversionsTest.cpp File Reference

#include <PrototxtConversions.hpp>
#include "armnn/Tensor.hpp"
#include <boost/test/unit_test.hpp>

Go to the source code of this file.

Functions
	BOOST_AUTO_TEST_CASE (ConvertInt32ToOctalStringTest)
	BOOST_AUTO_TEST_CASE (ConvertTensorShapeToStringTest)

Function Documentation

BOOST_AUTO_TEST_CASE() [1/2]

BOOST_AUTO_TEST_CASE

(

ConvertInt32ToOctalStringTest

)

Definition at line 13 of file PrototxtConversionsTest.cpp.

References armnnUtils::ConvertInt32ToOctalString().

{
    using armnnUtils::ConvertInt32ToOctalString;

    std::string octalString = ConvertInt32ToOctalString(1);
    BOOST_ASSERT(octalString.compare("\\\\001\\\\000\\\\000\\\\000"));

    octalString = ConvertInt32ToOctalString(256);
    BOOST_ASSERT(octalString.compare("\\\\000\\\\100\\\\000\\\\000"));

    octalString = ConvertInt32ToOctalString(65536);
    BOOST_ASSERT(octalString.compare("\\\\000\\\\000\\\\100\\\\000"));

    octalString = ConvertInt32ToOctalString(16777216);
    BOOST_ASSERT(octalString.compare("\\\\000\\\\000\\\\000\\\\100"));

    octalString = ConvertInt32ToOctalString(-1);
    BOOST_ASSERT(octalString.compare("\\\\377\\\\377\\\\377\\\\377"));

    octalString = ConvertInt32ToOctalString(-256);
    BOOST_ASSERT(octalString.compare("\\\\000\\\\377\\\\377\\\\377"));

    octalString = ConvertInt32ToOctalString(-65536);
    BOOST_ASSERT(octalString.compare("\\\\000\\\\000\\\\377\\\\377"));

    octalString = ConvertInt32ToOctalString(-16777216);
    BOOST_ASSERT(octalString.compare("\\\\000\\\\000\\\\000\\\\377"));
}

BOOST_AUTO_TEST_CASE() [2/2]

BOOST_AUTO_TEST_CASE

(

ConvertTensorShapeToStringTest

)

Definition at line 42 of file PrototxtConversionsTest.cpp.

References BOOST_AUTO_TEST_SUITE_END(), and armnnUtils::ConvertTensorShapeToString().

{
    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"
        "}"
        ));
}