X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=tests%2Fbenchmark%2Fptime;h=f0fec443b152e1dfa4fe54d4e319685394c16a7c;hb=c0c0989ab70574e09b2f7e8b48c2da6af664a849;hp=8e22c02a15256f5eed1ef26125f49fb597f27337;hpb=c80a4aaefb80e2f43ac474aeda2ce8db45e12799;p=lttng-ust.git

diff --git a/tests/benchmark/ptime b/tests/benchmark/ptime
index 8e22c02a..f0fec443 100755
--- a/tests/benchmark/ptime
+++ b/tests/benchmark/ptime
@@ -1,4 +1,6 @@
-#!/usr/bin/python
+#!/usr/bin/env python
+#
+# SPDX-License-Identifier: LGPL-2.1-only
 
 import sys
 import time
@@ -7,16 +9,16 @@ import os
 def main():
 	args = sys.argv[1:]
 	if len(args) < 1:
-		print "usage: %s COMMAND" % sys.argv[0]
+		print("usage: %s COMMAND" % sys.argv[0])
 		sys.exit(1)
 
-	cmd = args[0]
+	cmd = ' '.join(args)
 
 	t1 = time.time()
 	os.system(cmd)
-	1
 	t2 = time.time()
 
-	print (t2-t1)
+	print(t2-t1)
 
-main()
+if __name__ == "__main__":
+	main()