refactor(tests): reorganize test layout and include grouping

chen2021673 · chen2021673 · commit bd9dfe241372 · 2026-05-12T09:25:43.000Z
- Move gtest header into its own include group between stdlib and
  project headers across all test sources, per project convention
  that third-party headers sit separately above project includes.
- Split device-specific tests in tests/tensor/ into cpu_only/ and
  cuda_only/ subdirectories, each built as an independent test
  target; CUDA-only tests are skipped when USE_CUDA=OFF.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -216,3 +216,17 @@ set_target_properties(infini_run PROPERTIES RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BIN
 if(BUILD_TEST)
   add_subdirectory(tests)
 endif()
+
+# Negative compile test: missing dtype registration must fail at compile time.
+set(DTYPE_DISPATCH_COMPILE_FAIL_SOURCE
+  ${PROJECT_SOURCE_DIR}/tests/dtype/test_dtype_dispatch_compile_fail.cc)
+
+add_executable(test_dtype_dispatch_compile_fail EXCLUDE_FROM_ALL
+  ${DTYPE_DISPATCH_COMPILE_FAIL_SOURCE}
+)
+
+target_include_directories(test_dtype_dispatch_compile_fail PRIVATE
+  ${PROJECT_SOURCE_DIR}
+)
+
+link_infini_train_exe(test_dtype_dispatch_compile_fail)
diff --git a/docs/test_infrastructure_design.md b/docs/test_infrastructure_design.md
@@ -111,8 +111,10 @@ mkdir tests/foo
 
 ```cpp
 // tests/foo/test_foo_basic.cc
-#include <gtest/gtest.h>
+#include "gtest/gtest.h"
+
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
 
 using namespace infini_train;
diff --git a/tests/autograd/test_autograd.cc b/tests/autograd/test_autograd.cc
@@ -1,6 +1,8 @@
 #include <cmath>
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/activations.h"
 #include "infini_train/include/autograd/elementwise.h"
 #include "infini_train/include/autograd/function.h"
@@ -13,8 +15,8 @@
 #include "infini_train/include/autograd/softmax.h"
 #include "infini_train/include/autograd/transform.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_elementwise_backward.cc b/tests/autograd/test_autograd_elementwise_backward.cc
@@ -1,11 +1,13 @@
 #include <cmath>
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/elementwise.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_elementwise_forward.cc b/tests/autograd/test_autograd_elementwise_forward.cc
@@ -1,12 +1,14 @@
 #include <cmath>
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/activations.h"
 #include "infini_train/include/autograd/elementwise.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_linear_backward.cc b/tests/autograd/test_autograd_linear_backward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/linear.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_linear_forward.cc b/tests/autograd/test_autograd_linear_forward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/linear.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_matmul_backward.cc b/tests/autograd/test_autograd_matmul_backward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/matmul.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_matmul_forward.cc b/tests/autograd/test_autograd_matmul_forward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/matmul.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_normalization_backward.cc b/tests/autograd/test_autograd_normalization_backward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/normalization.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_normalization_forward.cc b/tests/autograd/test_autograd_normalization_forward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/normalization.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_reduction_backward.cc b/tests/autograd/test_autograd_reduction_backward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/reduction.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_reduction_forward.cc b/tests/autograd/test_autograd_reduction_forward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/reduction.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_softmax_backward.cc b/tests/autograd/test_autograd_softmax_backward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/softmax.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_softmax_forward.cc b/tests/autograd/test_autograd_softmax_forward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/softmax.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_transform_backward.cc b/tests/autograd/test_autograd_transform_backward.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/transform.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/autograd/test_autograd_transform_forward.cc b/tests/autograd/test_autograd_transform_forward.cc
@@ -1,11 +1,13 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/misc.h"
 #include "infini_train/include/autograd/transform.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/common/test_main.cc b/tests/common/test_main.cc
@@ -1,6 +1,7 @@
-#include "infini_train/include/nn/parallel/global.h"
 #include "gtest/gtest.h"
 
+#include "infini_train/include/nn/parallel/global.h"
+
 int main(int argc, char **argv) {
     ::testing::InitGoogleTest(&argc, argv);
     infini_train::nn::parallel::global::GlobalEnv::Instance().Init(1, 1, false, 1, 1);
diff --git a/tests/dtype/CMakeLists.txt b/tests/dtype/CMakeLists.txt
@@ -8,33 +8,3 @@ list(FILTER DTYPE_SOURCES EXCLUDE REGEX ".*compile_fail.*")
 infini_train_add_test_suite(test_dtype
   SOURCES ${DTYPE_SOURCES}
 )
-
-# Negative compile test: missing dtype registration must fail at compile time.
-set(DTYPE_DISPATCH_COMPILE_FAIL_SOURCE
-  ${CMAKE_CURRENT_SOURCE_DIR}/test_dtype_dispatch_compile_fail.cc)
-
-try_compile(DTYPE_DISPATCH_COMPILE_UNEXPECTEDLY_SUCCEEDED
-  ${CMAKE_BINARY_DIR}/CMakeFiles/try_compile_dtype_dispatch_missing_map
-  SOURCES ${DTYPE_DISPATCH_COMPILE_FAIL_SOURCE}
-  CMAKE_FLAGS
-    "-DCMAKE_CXX_STANDARD=${CMAKE_CXX_STANDARD}"
-    "-DCMAKE_CXX_STANDARD_REQUIRED=ON"
-    "-DCMAKE_CXX_EXTENSIONS=OFF"
-    "-DCMAKE_CXX_FLAGS=-I${PROJECT_SOURCE_DIR}"
-  OUTPUT_VARIABLE DTYPE_DISPATCH_TRY_COMPILE_OUTPUT
-)
-
-if(DTYPE_DISPATCH_COMPILE_UNEXPECTEDLY_SUCCEEDED)
-  message(FATAL_ERROR
-    "dtype dispatch compile-fail test unexpectedly succeeded.\n"
-    "Source: ${DTYPE_DISPATCH_COMPILE_FAIL_SOURCE}\n"
-    "Output:\n${DTYPE_DISPATCH_TRY_COMPILE_OUTPUT}")
-else()
-  message(STATUS "compile-fail output:\n${DTYPE_DISPATCH_TRY_COMPILE_OUTPUT}")
-endif()
-
-add_custom_target(test_dtype_dispatch_compile_fail
-  COMMAND ${CMAKE_COMMAND} -E echo
-    "dtype dispatch compile-fail check passed (missing dtype registration correctly fails to compile)."
-  VERBATIM
-)
diff --git a/tests/dtype/test_dtype_dispatch.cc b/tests/dtype/test_dtype_dispatch.cc
@@ -1,10 +1,12 @@
 #include <type_traits>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/datatype.h"
 #include "infini_train/include/dtype_dispatch.h"
+
 #include "infini_train/src/core/runtime/cpu/cpu_dispatch.h"
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/dtype/test_scalar.cc b/tests/dtype/test_scalar.cc
@@ -2,9 +2,11 @@
 #include <cstdint>
 #include <limits>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/scalar.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/hook/test_hook.cc b/tests/hook/test_hook.cc
@@ -1,14 +1,16 @@
 #include <memory>
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/autograd/elementwise.h"
 #include "infini_train/include/autograd/function.h"
 #include "infini_train/include/autograd/function_hook.h"
 #include "infini_train/include/common/hook.h"
 #include "infini_train/include/nn/modules/module.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/hook/test_precision_check.cc b/tests/hook/test_precision_check.cc
@@ -2,12 +2,14 @@
 #include <memory>
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/nn/modules/module.h"
 #include "infini_train/include/tensor.h"
 #include "infini_train/include/utils/precision_check_config.h"
 #include "infini_train/include/utils/precision_checker.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/lora/test_lora.cc b/tests/lora/test_lora.cc
@@ -2,6 +2,8 @@
 #include <memory>
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/nn/lora/lora_config.h"
 #include "infini_train/include/nn/lora/lora_linear.h"
 #include "infini_train/include/nn/lora/lora_utils.h"
@@ -10,8 +12,8 @@
 #include "infini_train/include/nn/modules/module.h"
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 using namespace infini_train::nn::lora;
diff --git a/tests/optimizer/test_optimizer_creation.cc b/tests/optimizer/test_optimizer_creation.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/optimizer.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/optimizer/test_optimizer_step.cc b/tests/optimizer/test_optimizer_step.cc
@@ -1,10 +1,12 @@
 #include <vector>
 
+#include "gtest/gtest.h"
+
 #include "infini_train/include/nn/parallel/global.h"
 #include "infini_train/include/optimizer.h"
 #include "infini_train/include/tensor.h"
+
 #include "tests/common/test_utils.h"
-#include "gtest/gtest.h"
 
 using namespace infini_train;
 
diff --git a/tests/tensor/CMakeLists.txt b/tests/tensor/CMakeLists.txt
diff --git a/tests/tensor/cpu_only/test_tensor_copy.cc b/tests/tensor/cpu_only/test_tensor_copy.cc
diff --git a/tests/tensor/cuda_only/test_tensor_copy.cc b/tests/tensor/cuda_only/test_tensor_copy.cc
diff --git a/tests/tensor/test_tensor.cc b/tests/tensor/test_tensor.cc
diff --git a/tests/tensor/test_tensor_copy.cc b/tests/tensor/test_tensor_copy.cc
diff --git a/tests/tensor/test_tensor_create.cc b/tests/tensor/test_tensor_create.cc
diff --git a/tests/tensor/test_tensor_delete.cc b/tests/tensor/test_tensor_delete.cc
diff --git a/tests/transformer/test_transformer_architecture.cc b/tests/transformer/test_transformer_architecture.cc
