diff options
Diffstat (limited to 'src/armnn/test/ProfilerTests.cpp')
-rw-r--r-- | src/armnn/test/ProfilerTests.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnn/test/ProfilerTests.cpp b/src/armnn/test/ProfilerTests.cpp index a0df3b6b62..21900ffb9a 100644 --- a/src/armnn/test/ProfilerTests.cpp +++ b/src/armnn/test/ProfilerTests.cpp @@ -8,7 +8,6 @@ #include <armnn/utility/IgnoreUnused.hpp> #include <boost/test/unit_test.hpp> -#include <boost/test/tools/output_test_stream.hpp> #include <memory> #include <thread> @@ -225,9 +224,9 @@ BOOST_AUTO_TEST_CASE(WriteEventResults) size_t eventSequenceSizeAfter = armnn::GetProfilerEventSequenceSize(profiler.get()); BOOST_TEST(eventSequenceSizeAfter == eventSequenceSizeBefore + 1); - boost::test_tools::output_test_stream output; + std::ostringstream output; profiler->AnalyzeEventsAndWriteResults(output); - BOOST_TEST(!output.is_empty(false)); + BOOST_TEST(!output.str().empty()); // output should contain event name 'test' BOOST_CHECK(output.str().find("test") != std::string::npos); |