diff --git a/test/horver_correlation_test.cc b/test/horver_correlation_test.cc
index 9a178cf..c8daff8 100644
--- a/test/horver_correlation_test.cc
+++ b/test/horver_correlation_test.cc
@@ -37,6 +37,7 @@
  public:
   virtual void SetUp() {
     data_buf = (int16_t *)aom_malloc(MAX_SB_SQUARE * sizeof(int16_t));
+    ASSERT_NE(data_buf, nullptr);
     target_func_ = GET_PARAM(0);
   }
   virtual void TearDown() { aom_free(data_buf); }
diff --git a/test/pickrst_test.cc b/test/pickrst_test.cc
index 68b6621..0ccd609 100644
--- a/test/pickrst_test.cc
+++ b/test/pickrst_test.cc
@@ -47,12 +47,16 @@
     target_func_ = GET_PARAM(0);
     src_ = (uint8_t *)(aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK *
                                   sizeof(uint8_t)));
+    ASSERT_NE(src_, nullptr);
     dgd_ = (uint8_t *)(aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK *
                                   sizeof(uint8_t)));
+    ASSERT_NE(dgd_, nullptr);
     flt0_ = (int32_t *)(aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK *
                                    sizeof(int32_t)));
+    ASSERT_NE(flt0_, nullptr);
     flt1_ = (int32_t *)(aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK *
                                    sizeof(int32_t)));
+    ASSERT_NE(flt1_, nullptr);
   }
   virtual void TearDown() {
     aom_free(src_);
@@ -210,12 +214,16 @@
     target_func_ = GET_PARAM(0);
     src_ =
         (uint16_t *)aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK * sizeof(*src_));
+    ASSERT_NE(src_, nullptr);
     dgd_ =
         (uint16_t *)aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK * sizeof(*dgd_));
+    ASSERT_NE(dgd_, nullptr);
     flt0_ =
         (int32_t *)aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK * sizeof(*flt0_));
+    ASSERT_NE(flt0_, nullptr);
     flt1_ =
         (int32_t *)aom_malloc(MAX_DATA_BLOCK * MAX_DATA_BLOCK * sizeof(*flt1_));
+    ASSERT_NE(flt1_, nullptr);
   }
   virtual void TearDown() {
     aom_free(src_);