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
