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