add argument passing via TEST_ARGS environment variable to all tests with an explicit main function

unittest/commands/test_simple_commands.cpp

@@ -26,6 +26,8 @@
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
 
+using LAMMPS_NS::utils::split_words;
+
 namespace LAMMPS_NS {
 using ::testing::Eq;
 
@@ -127,6 +129,17 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
+
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();

unittest/force-styles/test_main.cpp

@@ -82,15 +82,17 @@ int main(int argc, char **argv)
     }
 
     // handle arguments passed via environment variable
-    std::vector<std::string> env = split_words(getenv("TEST_ARGS"));
+    if (const char *var = getenv("TEST_ARGS")) {
-    for (auto arg : env) {
+        std::vector<std::string> env = split_words(var);
-        if (arg == "-u") {
+        for (auto arg : env) {
-            generate_yaml_file(argv[1], test_config);
+            if (arg == "-u") {
-            return 0;
+                generate_yaml_file(argv[1], test_config);
-        } else if (arg == "-s") {
+                return 0;
-            print_stats = true;
+            } else if (arg == "-s") {
-        } else if (arg == "-v") {
+                print_stats = true;
-            verbose = true;
+            } else if (arg == "-v") {
+                verbose = true;
+            }
         }
     }
 

unittest/formats/test_atom_styles.cpp

@@ -38,6 +38,8 @@
 
 #define GETIDX(i) lmp->atom->map(i)
 
+using LAMMPS_NS::utils::split_words;
+
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
 
@@ -1167,6 +1169,16 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();

unittest/formats/test_eim_potential_file_reader.cpp

@@ -23,6 +23,7 @@
 #include <mpi.h>
 
 using namespace LAMMPS_NS;
+using utils::split_words;
 
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
@@ -208,6 +209,17 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
+
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();

unittest/formats/test_potential_file_reader.cpp

@@ -34,6 +34,7 @@
 #include <mpi.h>
 
 using namespace LAMMPS_NS;
+using utils::split_words;
 
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
@@ -198,6 +199,16 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();