diff options
Diffstat (limited to 'src/armnn/test/FlowControl.cpp')
-rw-r--r-- | src/armnn/test/FlowControl.cpp | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/armnn/test/FlowControl.cpp b/src/armnn/test/FlowControl.cpp index 6198ca8a24..b0667a3778 100644 --- a/src/armnn/test/FlowControl.cpp +++ b/src/armnn/test/FlowControl.cpp @@ -51,8 +51,18 @@ BOOST_AUTO_TEST_CASE(ErrorOnLoadNetwork) // optimize the network std::vector<BackendId> backends = {Compute::CpuRef}; - IOptimizedNetworkPtr optNet = Optimize(*net, backends, runtime->GetDeviceSpec()); - BOOST_CHECK(!optNet); // Should have failed to optimise, as flow control is not yet implemented + std::vector<std::string> errMessages; + + try + { + Optimize(*net, backends, runtime->GetDeviceSpec(), OptimizerOptions(), errMessages); + BOOST_FAIL("Should have thrown an exception."); + } + catch (const InvalidArgumentException& e) + { + // Different exceptions are thrown on different backends + } + BOOST_TEST(errMessages.size() > 1); } BOOST_AUTO_TEST_SUITE_END() |