From d6568772be726acd7dc2fc3e592f101c77d690a6 Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Wed, 22 Jul 2020 12:46:51 +0100
Subject: IVGCVSW-5010 Add GetCapabilities to ITensorHandleFactory

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ie8acb9c729af4f95488aecf795f45ff12364f9ca
---
 src/backends/reference/test/RefTensorHandleTests.cpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'src/backends/reference/test/RefTensorHandleTests.cpp')

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)
-- 
cgit v1.2.1