2 files changed, 39 insertions, 0 deletions
diff --git a/src/backends/reference/test/CMakeLists.txt b/src/backends/reference/test/CMakeLists.txt
index d5ce3553f1..9d6572a4af 100644
--- a/src/backends/reference/test/CMakeLists.txt
+++ b/src/backends/reference/test/CMakeLists.txt
@@ -5,6 +5,7 @@
 
 list(APPEND armnnRefBackendUnitTests_sources
     ArgMinMaxTests.cpp
+    RefBackendTests.cpp
     RefCreateWorkloadTests.cpp
     RefDetectionPostProcessTests.cpp
     RefEndToEndTests.cpp
diff --git a/src/backends/reference/test/RefBackendTests.cpp b/src/backends/reference/test/RefBackendTests.cpp
new file mode 100644
index 0000000000..5154ea4c8f
--- /dev/null
+++ b/src/backends/reference/test/RefBackendTests.cpp
@@ -0,0 +1,38 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include <reference/RefBackend.hpp>
+#include <reference/RefTensorHandleFactory.hpp>
+
+#include <doctest/doctest.h>
+
+using namespace armnn;
+
+TEST_SUITE("RefBackendTests")
+{
+TEST_CASE("RefRegisterTensorHandleFactoriesMatchingImportFactoryId")
+{
+    auto refBackend = std::make_unique<RefBackend>();
+    TensorHandleFactoryRegistry registry;
+    refBackend->RegisterTensorHandleFactories(registry);
+
+    // When calling RegisterTensorHandleFactories, CopyAndImportFactoryPair is registered
+    // Get matching import factory id correctly
+    CHECK((registry.GetMatchingImportFactoryId(RefTensorHandleFactory::GetIdStatic()) ==
+           RefTensorHandleFactory::GetIdStatic()));
+}
+
+TEST_CASE("RefCreateWorkloadFactoryMatchingImportFactoryId")
+{
+    auto refBackend = std::make_unique<RefBackend>();
+    TensorHandleFactoryRegistry registry;
+    refBackend->CreateWorkloadFactory(registry);
+
+    // When calling CreateWorkloadFactory, CopyAndImportFactoryPair is registered
+    // Get matching import factory id correctly
+    CHECK((registry.GetMatchingImportFactoryId(RefTensorHandleFactory::GetIdStatic()) ==
+           RefTensorHandleFactory::GetIdStatic()));
+}
+}