Rearrange unit test files (#52)

apache · May 24, 2017 · 5c7f715 · 5c7f715
1 parent 9c6889a
commit 5c7f715
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tests/cpp/test_utils.h → tests/cpp/include/test_ndarray_utils.h b/tests/cpp/test_utils.h → tests/cpp/include/test_ndarray_utils.h
diff --git a/tests/cpp/operator/batchnorm_test.cc b/tests/cpp/operator/batchnorm_test.cc
@@ -1,7 +1,7 @@
 /*!
  * Copyright (c) 2017 by Contributors
  * \file batchnorm_test.cc
- * \brief operator unit test utility functions
+ * \brief batchnorm operator unit test utility functions
  * \author Chris Olivier
 */
 
@@ -865,8 +865,8 @@ TEST(BATCH_NORM, TestIterAll) {
           kwargs.push_back({ "cudnn_off", "True" });
         }
         for (TShape shape : shapes) {
-          for (int g1 = 0; g1 < 2U; ++g1) {
-            for (int g2 = 0; g2 < 2U; ++g2) {
+          for (int g1 = 0; g1 < 2; ++g1) {
+            for (int g2 = 0; g2 < 2; ++g2) {
               for (int type : v2_types) {
                 MSHADOW_REAL_TYPE_SWITCH_EX(
                   type, DType, AccReal,

diff --git a/tests/cpp/ndarray_test.cc → tests/cpp/operator/ndarray_test.cc b/tests/cpp/ndarray_test.cc → tests/cpp/operator/ndarray_test.cc
@@ -18,7 +18,7 @@
 #include "../src/operator/tensor/indexing_op.h"
 #include "../src/operator/optimizer_op-inl.h"
 #include "../src/operator/tensor/init_op.h"
-#include "test_utils.h"
+#include "test_ndarray_utils.h"
 
 using namespace mxnet;
 // Conversion Tests