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)
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_tracesize():
arrs = get_sample_arrays()
try:
multipletau.autocorrelate(a=arrs[0][:31], m=16)
except ValueError as e:
assert '`len(a)` must be >= `2m`!' in e.args
else:
assert False
def test_ac_trace0():
arrs = get_sample_arrays()
try:
multipletau.autocorrelate(a=arrs[0] - np.mean(arrs[0]), normalize=True)
except ValueError as e:
assert "Cannot normalize: Average of `a` is zero!" in e.args
else:
assert False
def test_numpy_cc_samesize():
arrs = get_sample_arrays()
try:
multipletau.correlate_numpy(a=arrs[0], v=arrs[1], normalize=True)
except ValueError as e:
assert "`a` and `v` must have same length!" in e.args
else:
assert False
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_cc():
arrs = get_sample_arrays()
try:
multipletau.correlate(a=arrs[0], v=arrs[0], copy=2)
except ValueError as e:
assert "`copy` must be boolean!" in e.args
else:
assert False
try:
multipletau.correlate(a=arrs[0], v=arrs[0], ret_sum=2)
except ValueError as e:
assert "`ret_sum` must be boolean!" in e.args
else:
assert False
try:
multipletau.correlate(a=arrs[0], v=arrs[0], normalize=2)
except ValueError as e:
assert "`normalize` must be boolean!" in e.args
else:
assert False
try:
multipletau.correlate(a=arrs[0], v=arrs[0], compress="peter")
except ValueError as e:
assert "Invalid value for `compress`!" in e.args[0]
else:
assert False
try:
multipletau.correlate(a=arrs[0],
v=arrs[0],
normalize=True,
ret_sum=True)
except ValueError as e:
assert "'normalize' and 'ret_sum' must not both be True!" in e.args
else:
assert False
