add testing on correlate.py