def testFromEventsList_ShiftRange(self):
pianoroll_seq = list(pianoroll_lib.PianorollSequence(
events_list=[(0, 1), (2, 3), (4, 5), (6,)], steps_per_quarter=1,
min_pitch=1, max_pitch=4, shift_range=True))
expected_pianoroll_seq = [
(0,),
(1, 2),
(3,),
(),
]
self.assertEqual(expected_pianoroll_seq, pianoroll_seq)
def testToSequence(self):
testing_lib.add_track_to_sequence(
self.note_sequence, 0,
[(60, 100, 0.0, 4.0), (64, 100, 0.0, 3.0), (67, 100, 0.0, 1.0),
(67, 100, 3.0, 4.0)])
quantized_sequence = sequences_lib.quantize_note_sequence(
self.note_sequence, steps_per_quarter=1)
pianoroll_seq = pianoroll_lib.PianorollSequence(quantized_sequence)
pianoroll_seq_ns = pianoroll_seq.to_sequence(qpm=60.0)
# Make comparison easier
pianoroll_seq_ns.notes.sort(key=lambda n: (n.start_time, n.pitch))
self.note_sequence.notes.sort(key=lambda n: (n.start_time, n.pitch))
self.assertEqual(self.note_sequence, pianoroll_seq_ns)
def testFromQuantizedNoteSequence(self):
testing_lib.add_track_to_sequence(
self.note_sequence, 0,
[(20, 100, 0.0, 4.0), (24, 100, 0.0, 1.0), (26, 100, 0.0, 3.0),
(110, 100, 1.0, 2.0), (24, 100, 2.0, 4.0)])
quantized_sequence = sequences_lib.quantize_note_sequence(
self.note_sequence, steps_per_quarter=1)
pianoroll_seq = list(pianoroll_lib.PianorollSequence(quantized_sequence))
expected_pianoroll_seq = [
(3, 5),
(5,),
(3, 5),
(3,),
]
self.assertEqual(expected_pianoroll_seq, pianoroll_seq)
def testSetLengthRemoveSteps(self):
pianoroll_seq = pianoroll_lib.PianorollSequence(steps_per_quarter=1)
pianoroll_events = [(), (2, 4), (2, 4), (2,), (5,)]
for event in pianoroll_events:
pianoroll_seq.append(event)
pianoroll_seq.set_length(2)
self.assertEqual([(), (2, 4)], list(pianoroll_seq))
pianoroll_seq.set_length(1)
self.assertEqual([()], list(pianoroll_seq))
pianoroll_seq.set_length(0)
self.assertEqual([], list(pianoroll_seq))
def testFromQuantizedNoteSequence_SplitRepeats(self):
testing_lib.add_track_to_sequence(
self.note_sequence, 0,
[(0, 100, 0.0, 2.0), (0, 100, 2.0, 4.0), (1, 100, 0.0, 2.0),
(2, 100, 2.0, 4.0)])
quantized_sequence = sequences_lib.quantize_note_sequence(
self.note_sequence, steps_per_quarter=1)
pianoroll_seq = list(pianoroll_lib.PianorollSequence(
quantized_sequence, min_pitch=0, split_repeats=True))
expected_pianoroll_seq = [
(0, 1),
(1,),
(0, 2),
(0, 2),
]
self.assertEqual(expected_pianoroll_seq, pianoroll_seq)
def testSetLengthAddSteps(self):
pianoroll_seq = pianoroll_lib.PianorollSequence(steps_per_quarter=1)
pianoroll_seq.append((0))
self.assertEqual(1, pianoroll_seq.num_steps)
pianoroll_seq.set_length(5)
self.assertEqual(5, pianoroll_seq.num_steps)
self.assertEqual([(0), (), (), (), ()], list(pianoroll_seq))
# Add 5 more steps.
pianoroll_seq.set_length(10)
self.assertEqual(10, pianoroll_seq.num_steps)
self.assertEqual([(0)] + [()] * 9, list(pianoroll_seq))
def testToSequenceWithBaseNoteSequence(self):
pianoroll_seq = pianoroll_lib.PianorollSequence(
steps_per_quarter=1, start_step=1)
pianoroll_events = [(39, 43), (39, 43)]
for event in pianoroll_events:
pianoroll_seq.append(event)
base_seq = copy.deepcopy(self.note_sequence)
testing_lib.add_track_to_sequence(
base_seq, 0, [(60, 100, 0.0, 1.0)])
pianoroll_seq_ns = pianoroll_seq.to_sequence(
qpm=60.0, base_note_sequence=base_seq)
testing_lib.add_track_to_sequence(
self.note_sequence, 0,
[(60, 100, 0.0, 1.0), (60, 100, 1.0, 3.0), (64, 100, 1.0, 3.0)])
# Make comparison easier
pianoroll_seq_ns.notes.sort(key=lambda n: (n.start_time, n.pitch))
self.note_sequence.notes.sort(key=lambda n: (n.start_time, n.pitch))
self.assertEqual(self.note_sequence, pianoroll_seq_ns)
