X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=tests%2Fbenchmark%2Fptime;h=419f3684d6b5373f5f8b39be3ab1ac1f1389ea88;hb=a44af49d539038fd64c124deaaf07a0c2f9ec039;hp=8e22c02a15256f5eed1ef26125f49fb597f27337;hpb=42f209e98d68936df04ec947a12e684155ce7cfe;p=lttng-ust.git

diff --git a/tests/benchmark/ptime b/tests/benchmark/ptime
index 8e22c02a..419f3684 100755
--- a/tests/benchmark/ptime
+++ b/tests/benchmark/ptime
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/env python
 
 import sys
 import time
@@ -7,16 +7,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()