Added times to the output of test.py

This is for benchmarking.