diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/reference/test/RefTensorHandleTests.cpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefTensorHandleTests.cpp b/src/backends/reference/test/RefTensorHandleTests.cpp index 42f5664b72..3635a32472 100644 --- a/src/backends/reference/test/RefTensorHandleTests.cpp +++ b/src/backends/reference/test/RefTensorHandleTests.cpp @@ -141,6 +141,23 @@ BOOST_AUTO_TEST_CASE(RefTensorHandleImport) BOOST_CHECK(buffer[1] == 10.0f); } +BOOST_AUTO_TEST_CASE(RefTensorHandleGetCapabilities) +{ + std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); + RefTensorHandleFactory handleFactory(memoryManager); + + // Builds up the structure of the network. + INetworkPtr net(INetwork::Create()); + IConnectableLayer* input = net->AddInputLayer(0); + IConnectableLayer* output = net->AddOutputLayer(0); + input->GetOutputSlot(0).Connect(output->GetInputSlot(0)); + + std::vector<Capability> capabilities = handleFactory.GetCapabilities(input, + output, + CapabilityClass::PaddingRequired); + BOOST_CHECK(capabilities.empty()); +} + #if !defined(__ANDROID__) // Only run these tests on non Android platforms BOOST_AUTO_TEST_CASE(CheckSourceType) |