diff options
Diffstat (limited to 'src/armnnUtils/test/QuantizeHelperTest.cpp')
-rw-r--r-- | src/armnnUtils/test/QuantizeHelperTest.cpp | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/src/armnnUtils/test/QuantizeHelperTest.cpp b/src/armnnUtils/test/QuantizeHelperTest.cpp new file mode 100644 index 0000000000..7e781d0b5d --- /dev/null +++ b/src/armnnUtils/test/QuantizeHelperTest.cpp @@ -0,0 +1,46 @@ +// +// Copyright © 2019 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <QuantizeHelper.hpp> + +#include <boost/core/ignore_unused.hpp> +#include <boost/test/unit_test.hpp> + +#include <vector> + +BOOST_AUTO_TEST_SUITE(QuantizeHelper) + +namespace +{ + +template<typename T> +bool IsFloatIterFunc(T iter) +{ + boost::ignore_unused(iter); + return armnnUtils::IsFloatingPointIterator<T>::value; +} + +} // anonymous namespace + +BOOST_AUTO_TEST_CASE(IsFloatIterFuncTest) +{ + std::vector<float> fArray; + BOOST_TEST(IsFloatIterFunc(fArray.begin()) == true); + BOOST_TEST(IsFloatIterFunc(fArray.cbegin()) == true); + + std::vector<double> dArray; + BOOST_TEST(IsFloatIterFunc(dArray.begin()) == true); + + std::vector<int> iArray; + BOOST_TEST(IsFloatIterFunc(iArray.begin()) == false); + + float floats[5]; + BOOST_TEST(IsFloatIterFunc(&floats[0]) == true); + + int ints[5]; + BOOST_TEST(IsFloatIterFunc(&ints[0]) == false); +} + +BOOST_AUTO_TEST_SUITE_END() |