1 files changed, 51 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefMemoryManagerTests.cpp b/src/backends/reference/test/RefMemoryManagerTests.cpp
new file mode 100644
index 0000000000..15b7c2af4f
--- /dev/null
+++ b/src/backends/reference/test/RefMemoryManagerTests.cpp
@@ -0,0 +1,51 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include <reference/RefMemoryManager.hpp>
+
+#include <boost/test/unit_test.hpp>
+
+BOOST_AUTO_TEST_SUITE(RefMemoryManagerTests)
+using namespace armnn;
+using Pool = RefMemoryManager::Pool;
+
+BOOST_AUTO_TEST_CASE(ManageOneThing)
+{
+    RefMemoryManager memoryManager;
+
+    Pool* pool = memoryManager.Manage(10);
+
+    BOOST_CHECK(pool);
+
+    memoryManager.Acquire();
+
+    BOOST_CHECK(memoryManager.GetPointer(pool) != nullptr); // Yields a valid pointer
+
+    memoryManager.Release();
+}
+
+BOOST_AUTO_TEST_CASE(ManageTwoThings)
+{
+    RefMemoryManager memoryManager;
+
+    Pool* pool1 = memoryManager.Manage(10);
+    Pool* pool2 = memoryManager.Manage(5);
+
+    BOOST_CHECK(pool1);
+    BOOST_CHECK(pool2);
+
+    memoryManager.Acquire();
+
+    void *p1 = memoryManager.GetPointer(pool1);
+    void *p2 = memoryManager.GetPointer(pool2);
+
+    BOOST_CHECK(p1);
+    BOOST_CHECK(p2);
+    BOOST_CHECK(p1 != p2);
+
+    memoryManager.Release();
+}
+
+BOOST_AUTO_TEST_SUITE_END()