def get_result_from_folder(output_folder): output_folder = Path(output_folder) sorting = se.HDSortSortingExtractor( file_path=str(output_folder / 'hdsort_output' / 'hdsort_output_results.mat')) return sorting
def test_hdsort_extractor(self): path = self.test_dir + '/results_test_hdsort_extractor.mat' locations = np.ones((10, 2)) se.HDSortSortingExtractor.write_sorting(self.SX, path, locations=locations, noise_std_by_channel=None) SX_hd = se.HDSortSortingExtractor(path) check_sorting_return_types(SX_hd) check_sortings_equal(self.SX, SX_hd) check_dumping(SX_hd)