maintain the integration of gtest arguments (#1491)

onnx · Oct 10, 2018 · 354fa9d · 354fa9d
1 parent 0c8dead
commit 354fa9d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/onnx/backend/test/cpp/test_main.cc b/onnx/backend/test/cpp/test_main.cc
@@ -2,11 +2,12 @@
 #include "gtest_utils.h"
 
 GTEST_API_ int main(int argc, char** argv) {
-  if (argc != 2) {
+  if (argc < 2) {
     std::cerr << "target directory must be given!" << std::endl;
     return EXIT_FAILURE;
   }
-  std::string target_dir = argv[1];
+  argc--;
+  std::string target_dir = argv[argc];
   auto testcases = ONNX_NAMESPACE::testing::LoadAllTestCases(target_dir);
   GetTestCases() = testcases;
   ::testing::InitGoogleTest(&argc, argv);