commit 2f885bbcb031dccfd97ab5fe8a152b987124c87b
author Georgia Kouveli <georgia.kouveli@arm.com> Thu Dec 15 18:43:42 2016 +0000
committer Georgia Kouveli <georgia.kouveli@arm.com> Fri Dec 16 15:37:13 2016 +0000
tree 6b32138c467742514385f0e2a748f20de2005fef
parent be9c4d02c96eaa6cdc9a9a5ae3cc65f57d54585c

Add option to skip generation of dummy traces in tools/generate_tests.py.

Change-Id: Ie5d16f2e6c430666bea256fd6344bbe03324fcc2

tools/generate_tests.py

diff --git a/tools/generate_tests.py b/tools/generate_tests.py
index 9831f85..342c21f 100755
--- a/tools/generate_tests.py
+++ b/tools/generate_tests.py
@@ -684,6 +684,8 @@
   result.add_argument('--jobs', '-j', type=int, metavar='N',
                       default=multiprocessing.cpu_count(),
                       help='Allow N jobs at once')
+  result.add_argument('--skip-traces', action='store_true',
+                      help='Skip generation of dummy traces.')
   return result.parse_args()
 
 
@@ -697,7 +699,7 @@
 // -----------------------------------------------------------------------------
     """.format(template_file)
 
-def GenerateTest(generator, clang_format):
+def GenerateTest(generator, clang_format, skip_traces):
   template_file = template_files[generator.test_type]
   generated_file = ""
   with open(template_file, "r") as f:
@@ -759,8 +761,9 @@
                             stdout=subprocess.PIPE)
     out, _ = proc.communicate(generated_file.encode())
     f.write(out.decode())
-  # Write dummy trace files into 'test/aarch32/traces/'.
-  generator.WriteEmptyTraces("test/aarch32/traces/")
+  if not skip_traces:
+    # Write dummy trace files into 'test/aarch32/traces/'.
+    generator.WriteEmptyTraces("test/aarch32/traces/")
   print("Generated {} test for \"{}\".".format(generator.test_type, generator.test_name))
 
 
@@ -774,5 +777,6 @@
   # Call the `GenerateTest` function for each generator object in parallel. This
   # will use as many processes as defined by `-jN`, which defaults to 1.
   with multiprocessing.Pool(processes=args.jobs) as pool:
-    pool.map(functools.partial(GenerateTest, clang_format=args.clang_format),
+    pool.map(functools.partial(GenerateTest, clang_format=args.clang_format,
+                                             skip_traces=args.skip_traces),
              generators)