def test_single_source_storage_opencl():
"""Test AIS estimation in MuTE example network."""
dat = Data()
dat.generate_mute_data(1000, 5)
max_lag = 5
analysis_opts = {
'cmi_calc_name': 'opencl_kraskov',
'n_perm_mi': 22,
'alpha_mi': 0.05,
'tail_mi': 'one',
}
processes = [1, 2, 3]
network_analysis = Single_process_storage(max_lag, analysis_opts, tau=1)
res = network_analysis.analyse_network(dat, processes)
print('AIS for MUTE data proc 1: {0}'.format(res[1]['ais']))
print('AIS for MUTE data proc 2: {0}'.format(res[2]['ais']))
print('AIS for MUTE data proc 3: {0}'.format(res[3]['ais']))
def test_single_source_storage_opencl():
"""Test AIS estimation in MuTE example network."""
dat = Data()
dat.generate_mute_data(1000, 5)
settings = {
'cmi_estimator': 'OpenCLKraskovCMI',
'max_lag': 5,
'tau': 1,
'n_perm_mi': 22,
'alpha_mi': 0.05,
'tail_mi': 'one',
}
processes = [1, 2, 3]
network_analysis = Single_process_storage()
res = network_analysis.analyse_network(settings, dat, processes)
print('AIS for MUTE data proc 1: {0}'.format(res[1]['ais']))
print('AIS for MUTE data proc 2: {0}'.format(res[2]['ais']))
print('AIS for MUTE data proc 3: {0}'.format(res[3]['ais']))
