commit b275da5afe480e994fe6cd897b3090c52f1bcdea
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri Dec 17 17:27:37 2021 +0000
committer Francis Murtagh <francis.murtagh@arm.com> Tue Jan 11 11:59:22 2022 +0000
tree 390246b981d5d39f79099b57a56c0f21c2932c05
parent 2048bcf8ed671b593ac9af2974e10319b9058b20

IVGCVSW-6675 Add functionality for CopyAndImportFactoryPair
 to TensorHandleFactoryRegistry

 * Add RegisterCopyAndImportFactoryPair
 * Add GetMatchingImportFactoryId
 * Unit tests

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I91e71bdeadec8fedbce7088016c06a47a03bdbaa

src/armnn/test/TensorHandleStrategyTest.cpp

diff --git a/src/armnn/test/TensorHandleStrategyTest.cpp b/src/armnn/test/TensorHandleStrategyTest.cpp
index fb26880..374479b 100644
--- a/src/armnn/test/TensorHandleStrategyTest.cpp
+++ b/src/armnn/test/TensorHandleStrategyTest.cpp
@@ -392,4 +392,18 @@
     CHECK(importCount == 1);
 }
 
+TEST_CASE("RegisterCopyAndImportFactoryPairTest")
+{
+    TensorHandleFactoryRegistry registry;
+    ITensorHandleFactory::FactoryId copyId = "CopyFactoryId";
+    ITensorHandleFactory::FactoryId importId = "ImportFactoryId";
+    registry.RegisterCopyAndImportFactoryPair(copyId, importId);
+
+    // Get mathing import factory id correctly
+    CHECK((registry.GetMatchingImportFactoryId(copyId) == importId));
+
+    // Return empty id when Invailid Id is given
+    CHECK((registry.GetMatchingImportFactoryId("InvalidFactoryId") == ""));
+}
+
 }