[Svace] Fix memory leak

This patch fixes memory leak.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>

diff --git a/tests/unittests/ne_core_data_test.cc b/tests/unittests/ne_core_data_test.cc
index 116695f..f42f4b4 100644 (file)
--- a/tests/unittests/ne_core_data_test.cc
+++ b/tests/unittests/ne_core_data_test.cc
@@ -400,10 +400,12 @@ TEST (ne_core_data_test, dummy_quantize) {
   uint32_t data_dims[] = {1, 1, 1, 1024};
 
   void *src_data = malloc (4096);
-  void *dst_data = malloc (4096);
-
   ASSERT_NE (src_data, nullptr);
-  ASSERT_NE (dst_data, nullptr);
+  void *dst_data = malloc (4096);
+  if (dst_data == nullptr) {
+    free (src_data);
+    ASSERT_NE (dst_data, nullptr);
+  }
 
   /* std to npu */
   std::unique_ptr<DataConverter> converter (new DataConverter (true));