rectified test.py and corrected benchmarking