def test_hs2_extractor(self): path1 = self.test_dir + '/firings_true.hdf5' se.HS2SortingExtractor.write_sorting(self.SX, path1) SX_hs2 = se.HS2SortingExtractor(path1) self._check_sorting_return_types(SX_hs2) self._check_sortings_equal(self.SX, SX_hs2) self.assertEqual(SX_hs2.get_sampling_frequency(), self.SX.get_sampling_frequency())
def get_result_from_folder(output_folder): return se.HS2SortingExtractor(output_folder / 'HS2_sorted.hdf5')
def test_hs2_extractor(self): path1 = self.test_dir + '/firings_true.hdf5' se.HS2SortingExtractor.writeSorting(self.SX, path1) SX_hs2 = se.HS2SortingExtractor(path1) self._check_sorting_return_types(SX_hs2) self._check_sortings_equal(self.SX, SX_hs2)
def get_result_from_folder(output_folder): return se.HS2SortingExtractor(file_path=Path(output_folder) / 'HS2_sorted.hdf5', load_unit_info=True)