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)
