def test_rectify(): rec, sort = se.example_datasets.toy_example(dump_folder='test', dumpable=True, duration=2, num_channels=4, seed=0) rec_rect = rectify(rec) assert np.allclose(rec_rect.get_traces(), np.abs(rec.get_traces())) check_dumping(rec_rect) shutil.rmtree('test')
def test_rectify(): rec, sort = se.example_datasets.create_dumpable_extractors(duration=10, num_channels=4, folder='test') rec_rect = rectify(rec) assert np.allclose(rec_rect.get_traces(), np.abs(rec.get_traces())) check_dumping(rec_rect) shutil.rmtree('test')
def test_rectify(): rec, sort = se.example_datasets.toy_example(duration=10, num_channels=4) rec_rect = rectify(rec) assert np.allclose(rec_rect.get_traces(), np.abs(rec.get_traces()))