Module note_seq.lead_sheets_lib_test
Tests for lead_sheets.
Expand source code
# Copyright 2021 The Magenta Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests for lead_sheets."""
import copy
from absl.testing import absltest
from note_seq import chords_lib
from note_seq import constants
from note_seq import lead_sheets_lib
from note_seq import melodies_lib
from note_seq import testing_lib
NOTE_OFF = constants.MELODY_NOTE_OFF
NO_EVENT = constants.MELODY_NO_EVENT
NO_CHORD = constants.NO_CHORD
class LeadSheetsLibTest(testing_lib.ProtoTestCase):
def testTranspose(self):
# LeadSheet transposition should agree with melody & chords transpositions.
melody_events = [12 * 5 + 4, NO_EVENT, 12 * 5 + 5,
NOTE_OFF, 12 * 6, NO_EVENT]
chord_events = [NO_CHORD, 'C', 'F', 'Dm', 'D', 'G']
melody = melodies_lib.Melody(melody_events)
chords = chords_lib.ChordProgression(chord_events)
expected_melody = copy.deepcopy(melody)
expected_chords = copy.deepcopy(chords)
lead_sheet = lead_sheets_lib.LeadSheet(melody, chords)
lead_sheet.transpose(transpose_amount=-5, min_note=12 * 5, max_note=12 * 7)
expected_melody.transpose(
transpose_amount=-5, min_note=12 * 5, max_note=12 * 7)
expected_chords.transpose(transpose_amount=-5)
self.assertEqual(expected_melody, lead_sheet.melody)
self.assertEqual(expected_chords, lead_sheet.chords)
def testSquash(self):
# LeadSheet squash should agree with melody squash & chords transpose.
melody_events = [12 * 5, NO_EVENT, 12 * 5 + 2,
NOTE_OFF, 12 * 6 + 4, NO_EVENT]
chord_events = ['C', 'Am', 'Dm', 'G', 'C', NO_CHORD]
melody = melodies_lib.Melody(melody_events)
chords = chords_lib.ChordProgression(chord_events)
expected_melody = copy.deepcopy(melody)
expected_chords = copy.deepcopy(chords)
lead_sheet = lead_sheets_lib.LeadSheet(melody, chords)
lead_sheet.squash(min_note=12 * 5, max_note=12 * 6, transpose_to_key=0)
transpose_amount = expected_melody.squash(
min_note=12 * 5, max_note=12 * 6, transpose_to_key=0)
expected_chords.transpose(transpose_amount=transpose_amount)
self.assertEqual(expected_melody, lead_sheet.melody)
self.assertEqual(expected_chords, lead_sheet.chords)
def testSetLength(self):
# Setting LeadSheet length should agree with setting length on melody and
# chords separately.
melody_events = [60]
chord_events = ['C7']
melody = melodies_lib.Melody(melody_events, start_step=9)
chords = chords_lib.ChordProgression(chord_events, start_step=9)
expected_melody = copy.deepcopy(melody)
expected_chords = copy.deepcopy(chords)
lead_sheet = lead_sheets_lib.LeadSheet(melody, chords)
lead_sheet.set_length(5)
expected_melody.set_length(5)
expected_chords.set_length(5)
self.assertEqual(expected_melody, lead_sheet.melody)
self.assertEqual(expected_chords, lead_sheet.chords)
self.assertEqual(9, lead_sheet.start_step)
self.assertEqual(14, lead_sheet.end_step)
self.assertListEqual([9, 10, 11, 12, 13], lead_sheet.steps)
def testToSequence(self):
# Sequence produced from lead sheet should contain notes from melody
# sequence and chords from chord sequence as text annotations.
melody = melodies_lib.Melody(
[NO_EVENT, 1, NO_EVENT, NOTE_OFF, NO_EVENT, 2, 3, NOTE_OFF, NO_EVENT])
chords = chords_lib.ChordProgression(
[NO_CHORD, 'A', 'A', 'C#m', 'C#m', 'D', 'B', 'B', 'B'])
lead_sheet = lead_sheets_lib.LeadSheet(melody, chords)
sequence = lead_sheet.to_sequence(
velocity=10,
instrument=1,
sequence_start_time=2,
qpm=60.0)
melody_sequence = melody.to_sequence(
velocity=10,
instrument=1,
sequence_start_time=2,
qpm=60.0)
chords_sequence = chords.to_sequence(
sequence_start_time=2,
qpm=60.0)
self.assertEqual(melody_sequence.ticks_per_quarter,
sequence.ticks_per_quarter)
self.assertProtoEquals(melody_sequence.tempos, sequence.tempos)
self.assertEqual(melody_sequence.total_time, sequence.total_time)
self.assertProtoEquals(melody_sequence.notes, sequence.notes)
self.assertProtoEquals(chords_sequence.text_annotations,
sequence.text_annotations)
if __name__ == '__main__':
absltest.main()Classes
class LeadSheetsLibTest (*args, **kwargs)-
Adds assertProtoEquals from tf.test.TestCase.
Create an instance of the class that will use the named test method when executed. Raises a ValueError if the instance does not have a method with the specified name.
Expand source code
class LeadSheetsLibTest(testing_lib.ProtoTestCase): def testTranspose(self): # LeadSheet transposition should agree with melody & chords transpositions. melody_events = [12 * 5 + 4, NO_EVENT, 12 * 5 + 5, NOTE_OFF, 12 * 6, NO_EVENT] chord_events = [NO_CHORD, 'C', 'F', 'Dm', 'D', 'G'] melody = melodies_lib.Melody(melody_events) chords = chords_lib.ChordProgression(chord_events) expected_melody = copy.deepcopy(melody) expected_chords = copy.deepcopy(chords) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) lead_sheet.transpose(transpose_amount=-5, min_note=12 * 5, max_note=12 * 7) expected_melody.transpose( transpose_amount=-5, min_note=12 * 5, max_note=12 * 7) expected_chords.transpose(transpose_amount=-5) self.assertEqual(expected_melody, lead_sheet.melody) self.assertEqual(expected_chords, lead_sheet.chords) def testSquash(self): # LeadSheet squash should agree with melody squash & chords transpose. melody_events = [12 * 5, NO_EVENT, 12 * 5 + 2, NOTE_OFF, 12 * 6 + 4, NO_EVENT] chord_events = ['C', 'Am', 'Dm', 'G', 'C', NO_CHORD] melody = melodies_lib.Melody(melody_events) chords = chords_lib.ChordProgression(chord_events) expected_melody = copy.deepcopy(melody) expected_chords = copy.deepcopy(chords) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) lead_sheet.squash(min_note=12 * 5, max_note=12 * 6, transpose_to_key=0) transpose_amount = expected_melody.squash( min_note=12 * 5, max_note=12 * 6, transpose_to_key=0) expected_chords.transpose(transpose_amount=transpose_amount) self.assertEqual(expected_melody, lead_sheet.melody) self.assertEqual(expected_chords, lead_sheet.chords) def testSetLength(self): # Setting LeadSheet length should agree with setting length on melody and # chords separately. melody_events = [60] chord_events = ['C7'] melody = melodies_lib.Melody(melody_events, start_step=9) chords = chords_lib.ChordProgression(chord_events, start_step=9) expected_melody = copy.deepcopy(melody) expected_chords = copy.deepcopy(chords) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) lead_sheet.set_length(5) expected_melody.set_length(5) expected_chords.set_length(5) self.assertEqual(expected_melody, lead_sheet.melody) self.assertEqual(expected_chords, lead_sheet.chords) self.assertEqual(9, lead_sheet.start_step) self.assertEqual(14, lead_sheet.end_step) self.assertListEqual([9, 10, 11, 12, 13], lead_sheet.steps) def testToSequence(self): # Sequence produced from lead sheet should contain notes from melody # sequence and chords from chord sequence as text annotations. melody = melodies_lib.Melody( [NO_EVENT, 1, NO_EVENT, NOTE_OFF, NO_EVENT, 2, 3, NOTE_OFF, NO_EVENT]) chords = chords_lib.ChordProgression( [NO_CHORD, 'A', 'A', 'C#m', 'C#m', 'D', 'B', 'B', 'B']) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) sequence = lead_sheet.to_sequence( velocity=10, instrument=1, sequence_start_time=2, qpm=60.0) melody_sequence = melody.to_sequence( velocity=10, instrument=1, sequence_start_time=2, qpm=60.0) chords_sequence = chords.to_sequence( sequence_start_time=2, qpm=60.0) self.assertEqual(melody_sequence.ticks_per_quarter, sequence.ticks_per_quarter) self.assertProtoEquals(melody_sequence.tempos, sequence.tempos) self.assertEqual(melody_sequence.total_time, sequence.total_time) self.assertProtoEquals(melody_sequence.notes, sequence.notes) self.assertProtoEquals(chords_sequence.text_annotations, sequence.text_annotations)Ancestors
- ProtoTestCase
- absl.testing.absltest.TestCase
- absl.third_party.unittest3_backport.case.TestCase
- unittest.case.TestCase
Methods
def testSetLength(self)-
Expand source code
def testSetLength(self): # Setting LeadSheet length should agree with setting length on melody and # chords separately. melody_events = [60] chord_events = ['C7'] melody = melodies_lib.Melody(melody_events, start_step=9) chords = chords_lib.ChordProgression(chord_events, start_step=9) expected_melody = copy.deepcopy(melody) expected_chords = copy.deepcopy(chords) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) lead_sheet.set_length(5) expected_melody.set_length(5) expected_chords.set_length(5) self.assertEqual(expected_melody, lead_sheet.melody) self.assertEqual(expected_chords, lead_sheet.chords) self.assertEqual(9, lead_sheet.start_step) self.assertEqual(14, lead_sheet.end_step) self.assertListEqual([9, 10, 11, 12, 13], lead_sheet.steps) def testSquash(self)-
Expand source code
def testSquash(self): # LeadSheet squash should agree with melody squash & chords transpose. melody_events = [12 * 5, NO_EVENT, 12 * 5 + 2, NOTE_OFF, 12 * 6 + 4, NO_EVENT] chord_events = ['C', 'Am', 'Dm', 'G', 'C', NO_CHORD] melody = melodies_lib.Melody(melody_events) chords = chords_lib.ChordProgression(chord_events) expected_melody = copy.deepcopy(melody) expected_chords = copy.deepcopy(chords) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) lead_sheet.squash(min_note=12 * 5, max_note=12 * 6, transpose_to_key=0) transpose_amount = expected_melody.squash( min_note=12 * 5, max_note=12 * 6, transpose_to_key=0) expected_chords.transpose(transpose_amount=transpose_amount) self.assertEqual(expected_melody, lead_sheet.melody) self.assertEqual(expected_chords, lead_sheet.chords) def testToSequence(self)-
Expand source code
def testToSequence(self): # Sequence produced from lead sheet should contain notes from melody # sequence and chords from chord sequence as text annotations. melody = melodies_lib.Melody( [NO_EVENT, 1, NO_EVENT, NOTE_OFF, NO_EVENT, 2, 3, NOTE_OFF, NO_EVENT]) chords = chords_lib.ChordProgression( [NO_CHORD, 'A', 'A', 'C#m', 'C#m', 'D', 'B', 'B', 'B']) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) sequence = lead_sheet.to_sequence( velocity=10, instrument=1, sequence_start_time=2, qpm=60.0) melody_sequence = melody.to_sequence( velocity=10, instrument=1, sequence_start_time=2, qpm=60.0) chords_sequence = chords.to_sequence( sequence_start_time=2, qpm=60.0) self.assertEqual(melody_sequence.ticks_per_quarter, sequence.ticks_per_quarter) self.assertProtoEquals(melody_sequence.tempos, sequence.tempos) self.assertEqual(melody_sequence.total_time, sequence.total_time) self.assertProtoEquals(melody_sequence.notes, sequence.notes) self.assertProtoEquals(chords_sequence.text_annotations, sequence.text_annotations) def testTranspose(self)-
Expand source code
def testTranspose(self): # LeadSheet transposition should agree with melody & chords transpositions. melody_events = [12 * 5 + 4, NO_EVENT, 12 * 5 + 5, NOTE_OFF, 12 * 6, NO_EVENT] chord_events = [NO_CHORD, 'C', 'F', 'Dm', 'D', 'G'] melody = melodies_lib.Melody(melody_events) chords = chords_lib.ChordProgression(chord_events) expected_melody = copy.deepcopy(melody) expected_chords = copy.deepcopy(chords) lead_sheet = lead_sheets_lib.LeadSheet(melody, chords) lead_sheet.transpose(transpose_amount=-5, min_note=12 * 5, max_note=12 * 7) expected_melody.transpose( transpose_amount=-5, min_note=12 * 5, max_note=12 * 7) expected_chords.transpose(transpose_amount=-5) self.assertEqual(expected_melody, lead_sheet.melody) self.assertEqual(expected_chords, lead_sheet.chords)
Inherited members