def test_remove_artifacts():
rec, sort = se.example_datasets.toy_example(dump_folder='test', dumpable=True, duration=2, num_channels=4, seed=0)
triggers = [15000, 30000]
ms = 10
ms_frames = int(ms * rec.get_sampling_frequency() / 1000)
traces_all_0_clean = rec.get_traces(start_frame=triggers[0] - ms_frames, end_frame=triggers[0] + ms_frames)
traces_all_1_clean = rec.get_traces(start_frame=triggers[1] - ms_frames, end_frame=triggers[1] + ms_frames)
rec_rmart = remove_artifacts(rec, triggers, ms_before=10, ms_after=10)
traces_all_0 = rec_rmart.get_traces(start_frame=triggers[0] - ms_frames, end_frame=triggers[0] + ms_frames)
traces_short_0 = rec_rmart.get_traces(start_frame=triggers[0] - 10, end_frame=triggers[0] + 10)
traces_all_1 = rec_rmart.get_traces(start_frame=triggers[1] - ms_frames, end_frame=triggers[1] + ms_frames)
traces_short_1 = rec_rmart.get_traces(start_frame=triggers[1] - 10, end_frame=triggers[1] + 10)
assert not np.any(traces_all_0)
assert not np.any(traces_all_1)
assert not np.any(traces_short_0)
assert not np.any(traces_short_1)
rec_rmart_lin = remove_artifacts(rec, triggers, ms_before=10, ms_after=10, mode="linear")
traces_all_0 = rec_rmart_lin.get_traces(start_frame=triggers[0] - ms_frames, end_frame=triggers[0] + ms_frames)
traces_all_1 = rec_rmart_lin.get_traces(start_frame=triggers[1] - ms_frames, end_frame=triggers[1] + ms_frames)
assert not np.allclose(traces_all_0, traces_all_0_clean)
assert not np.allclose(traces_all_1, traces_all_1_clean)
rec_rmart_cub = remove_artifacts(rec, triggers, ms_before=10, ms_after=10, mode="cubic")
traces_all_0 = rec_rmart_cub.get_traces(start_frame=triggers[0] - ms_frames, end_frame=triggers[0] + ms_frames)
traces_all_1 = rec_rmart_cub.get_traces(start_frame=triggers[1] - ms_frames, end_frame=triggers[1] + ms_frames)
assert not np.allclose(traces_all_0, traces_all_0_clean)
assert not np.allclose(traces_all_1, traces_all_1_clean)
check_dumping(rec_rmart)
shutil.rmtree('test')
def test_remove_artifacts():
rec, sort = se.example_datasets.create_dumpable_extractors(duration=10,
num_channels=4,
folder='test')
triggers = [30000, 90000]
ms = 10
ms_frames = int(ms * rec.get_sampling_frequency() / 1000)
rec_rmart = remove_artifacts(rec, triggers, ms_before=10, ms_after=10)
traces_all_0 = rec_rmart.get_traces(start_frame=triggers[0] - ms_frames,
end_frame=triggers[0] + ms_frames)
traces_short_0 = rec_rmart.get_traces(start_frame=triggers[0] - 10,
end_frame=triggers[0] + 10)
traces_all_1 = rec_rmart.get_traces(start_frame=triggers[1] - ms_frames,
end_frame=triggers[1] + ms_frames)
traces_short_1 = rec_rmart.get_traces(start_frame=triggers[1] - 10,
end_frame=triggers[1] + 10)
assert not np.any(traces_all_0)
assert not np.any(traces_all_1)
assert not np.any(traces_short_0)
assert not np.any(traces_short_1)
check_dumping(rec_rmart)
shutil.rmtree('test')
def test_remove_artifacts():
rec, sort = se.example_datasets.toy_example(duration=10, num_channels=4)
triggers = [30000, 90000]
ms = 10
ms_frames = int(ms * rec.get_sampling_frequency() / 1000)
rec_rmart = remove_artifacts(rec, triggers, ms_before=10, ms_after=10)
traces_all_0 = rec_rmart.get_traces(start_frame=triggers[0] - ms_frames,
end_frame=triggers[0] + ms_frames)
traces_short_0 = rec_rmart.get_traces(start_frame=triggers[0] - 10,
end_frame=triggers[0] + 10)
traces_all_1 = rec_rmart.get_traces(start_frame=triggers[1] - ms_frames,
end_frame=triggers[1] + ms_frames)
traces_short_1 = rec_rmart.get_traces(start_frame=triggers[1] - 10,
end_frame=triggers[1] + 10)
assert not np.any(traces_all_0)
assert not np.any(traces_all_1)
assert not np.any(traces_short_0)
assert not np.any(traces_short_1)