1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
""" Tests correlation-autocorrelation identity
"""
from __future__ import division, print_function
import numpy as np
import os
from os.path import abspath, basename, dirname, join, split, exists
import platform
import sys
import warnings
import zipfile
# Add parent directory to beginning of path variable
DIR = dirname(abspath(__file__))
sys.path = [split(DIR)[0]] + sys.path
import multipletau
from test_autocorrelate import get_sample_arrays
def test_ac_cc_m():
myframe = sys._getframe()
myname = myframe.f_code.co_name
print("running ", myname)
arrs = get_sample_arrays()
ms = [8, 16, 32, 64, 128]
a = np.concatenate(arrs)
res = []
for m in ms:
r = multipletau.autocorrelate(a=a,
m=m,
deltat=1,
normalize=False,
copy=True,
dtype=np.float_)
res.append(r)
res = np.concatenate(res)
rescc = []
for m in ms:
r = multipletau.correlate(a=a, v=a,
m=m,
deltat=1,
normalize=False,
copy=True,
dtype=np.float_)
rescc.append(r)
# test minimal length of array
_r2 = multipletau.correlate(a=a[:2*m], v=a[:2*m],
m=m,
deltat=1,
normalize=False,
copy=True,
dtype=np.float_)
rescc = np.concatenate(rescc)
assert np.all(res==rescc)
def test_ac_cc_normalize():
myframe = sys._getframe()
myname = myframe.f_code.co_name
print("running ", myname)
arrs = get_sample_arrays()
res = []
for a in arrs:
r = multipletau.autocorrelate(a=a,
m=16,
deltat=1,
normalize=True,
copy=True,
dtype=np.float_)
res.append(r)
res = np.concatenate(res)
rescc = []
for a in arrs:
r = multipletau.correlate(a=a, v=a,
m=16,
deltat=1,
normalize=True,
copy=True,
dtype=np.float_)
rescc.append(r)
rescc = np.concatenate(rescc)
assert np.all(res==rescc)
def test_ac_cc_simple():
myframe = sys._getframe()
myname = myframe.f_code.co_name
print("running ", myname)
arrs = get_sample_arrays()
rescc = []
for a in arrs:
r = multipletau.correlate(a=a, v=a,
m=16,
deltat=1,
normalize=False,
copy=True,
dtype=np.float_)
rescc.append(r)
rescc = np.concatenate(rescc)
resac = []
for a in arrs:
r = multipletau.autocorrelate(a=a,
m=16,
deltat=1,
normalize=False,
copy=True,
dtype=np.float_)
resac.append(r)
resac = np.concatenate(resac)
assert np.all(resac==rescc)
if __name__ == "__main__":
# Run all tests
loc = locals()
for key in list(loc.keys()):
if key.startswith("test_") and hasattr(loc[key], "__call__"):
loc[key]()
|