def test_merge():
scr1 = create_scores()[0]
scr1.sort()
scr2 = TrialScores(scr1.model_set[:10], scr1.seg_set, scr1.scores[:10, :],
scr1.score_mask[:10, :])
scr3 = TrialScores(scr1.model_set[10:], scr1.seg_set, scr1.scores[10:, :],
scr1.score_mask[10:, :])
scr4 = TrialScores.merge([scr2, scr3])
assert (scr1 == scr4)
scr2 = TrialScores(scr1.model_set, scr1.seg_set[:10], scr1.scores[:, :10],
scr1.score_mask[:, :10])
scr3 = TrialScores(scr1.model_set, scr1.seg_set[10:], scr1.scores[:, 10:],
scr1.score_mask[:, 10:])
scr4 = TrialScores.merge([scr2, scr3])
assert (scr1 == scr4)
def test_split():
scr1 = create_scores()[0]
scr1.sort()
num_parts = 3
scr_list = []
for i in xrange(num_parts):
for j in xrange(num_parts):
scr_ij = scr1.split(i + 1, num_parts, j + 1, num_parts)
scr_list.append(scr_ij)
scr2 = TrialScores.merge(scr_list)
assert (scr1 == scr2)
