Rewrite test_correlation_functions.py for new structure (#33)