diff options
Diffstat (limited to 'src/backends/reference/test/RefMemoryManagerTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefMemoryManagerTests.cpp | 25 |
1 files changed, 13 insertions, 12 deletions
diff --git a/src/backends/reference/test/RefMemoryManagerTests.cpp b/src/backends/reference/test/RefMemoryManagerTests.cpp index 15b7c2af4f..960e7ccad5 100644 --- a/src/backends/reference/test/RefMemoryManagerTests.cpp +++ b/src/backends/reference/test/RefMemoryManagerTests.cpp @@ -5,47 +5,48 @@ #include <reference/RefMemoryManager.hpp> -#include <boost/test/unit_test.hpp> +#include <doctest/doctest.h> -BOOST_AUTO_TEST_SUITE(RefMemoryManagerTests) +TEST_SUITE("RefMemoryManagerTests") +{ using namespace armnn; using Pool = RefMemoryManager::Pool; -BOOST_AUTO_TEST_CASE(ManageOneThing) +TEST_CASE("ManageOneThing") { RefMemoryManager memoryManager; Pool* pool = memoryManager.Manage(10); - BOOST_CHECK(pool); + CHECK(pool); memoryManager.Acquire(); - BOOST_CHECK(memoryManager.GetPointer(pool) != nullptr); // Yields a valid pointer + CHECK(memoryManager.GetPointer(pool) != nullptr); // Yields a valid pointer memoryManager.Release(); } -BOOST_AUTO_TEST_CASE(ManageTwoThings) +TEST_CASE("ManageTwoThings") { RefMemoryManager memoryManager; Pool* pool1 = memoryManager.Manage(10); Pool* pool2 = memoryManager.Manage(5); - BOOST_CHECK(pool1); - BOOST_CHECK(pool2); + CHECK(pool1); + CHECK(pool2); memoryManager.Acquire(); void *p1 = memoryManager.GetPointer(pool1); void *p2 = memoryManager.GetPointer(pool2); - BOOST_CHECK(p1); - BOOST_CHECK(p2); - BOOST_CHECK(p1 != p2); + CHECK(p1); + CHECK(p2); + CHECK(p1 != p2); memoryManager.Release(); } -BOOST_AUTO_TEST_SUITE_END() +} |