def test_cut_trials_larger_length(self): data = [self.data2[0]] seqs = gpfa_util.get_seqs(data, bin_size=self.bin_size) seg_length = seqs[0]['T'] + 1 with self.assertWarns(UserWarning): gpfa_util.cut_trials(seqs, seg_length=seg_length)
def test_cut_trials_zero_length(self): seqs = gpfa_util.get_seqs(self.data2, bin_size=self.bin_size) with self.assertRaises(ValueError): gpfa_util.cut_trials(seqs, seg_length=0)
def test_cut_trials_same_length(self): data = [self.data2[0]] seqs = gpfa_util.get_seqs(data, bin_size=self.bin_size) seg_length = seqs[0]['T'] seqs_cut = gpfa_util.cut_trials(seqs, seg_length=seg_length) assert_array_almost_equal(seqs[0]['y'], seqs_cut[0]['y'])
def test_cut_trials_inf(self): same_data = gpfa_util.cut_trials(self.data2, seg_length=np.Inf) assert same_data is self.data2