Module note_seq.chords_encoder_decoder_test
Tests for chords_encoder_decoder.
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 chords_encoder_decoder."""
from absl.testing import absltest
from note_seq import chords_encoder_decoder
from note_seq import constants
NO_CHORD = constants.NO_CHORD
class MajorMinorChordOneHotEncodingTest(absltest.TestCase):
def setUp(self):
self.enc = chords_encoder_decoder.MajorMinorChordOneHotEncoding()
def testEncodeNoChord(self):
index = self.enc.encode_event(NO_CHORD)
self.assertEqual(0, index)
def testEncodeChord(self):
# major triad
index = self.enc.encode_event('C')
self.assertEqual(1, index)
# minor triad
index = self.enc.encode_event('Cm')
self.assertEqual(13, index)
# dominant 7th
index = self.enc.encode_event('F7')
self.assertEqual(6, index)
# minor 9th
index = self.enc.encode_event('Abm9')
self.assertEqual(21, index)
def testEncodeThirdlessChord(self):
# suspended chord
with self.assertRaises(chords_encoder_decoder.ChordEncodingError):
self.enc.encode_event('Gsus4')
# power chord
with self.assertRaises(chords_encoder_decoder.ChordEncodingError):
self.enc.encode_event('Bb5')
def testDecodeNoChord(self):
figure = self.enc.decode_event(0)
self.assertEqual(NO_CHORD, figure)
def testDecodeChord(self):
# major chord
figure = self.enc.decode_event(3)
self.assertEqual('D', figure)
# minor chord
figure = self.enc.decode_event(17)
self.assertEqual('Em', figure)
class TriadChordOneHotEncodingTest(absltest.TestCase):
def setUp(self):
self.enc = chords_encoder_decoder.TriadChordOneHotEncoding()
def testEncodeNoChord(self):
index = self.enc.encode_event(NO_CHORD)
self.assertEqual(0, index)
def testEncodeChord(self):
# major triad
index = self.enc.encode_event('C13')
self.assertEqual(1, index)
# minor triad
index = self.enc.encode_event('Cm(maj7)')
self.assertEqual(13, index)
# augmented triad
index = self.enc.encode_event('Faug7')
self.assertEqual(30, index)
# diminished triad
index = self.enc.encode_event('Abm7b5')
self.assertEqual(45, index)
def testEncodeThirdlessChord(self):
# suspended chord
with self.assertRaises(chords_encoder_decoder.ChordEncodingError):
self.enc.encode_event('Gsus4')
# power chord
with self.assertRaises(chords_encoder_decoder.ChordEncodingError):
self.enc.encode_event('Bb5')
def testDecodeNoChord(self):
figure = self.enc.decode_event(0)
self.assertEqual(NO_CHORD, figure)
def testDecodeChord(self):
# major chord
figure = self.enc.decode_event(3)
self.assertEqual('D', figure)
# minor chord
figure = self.enc.decode_event(17)
self.assertEqual('Em', figure)
# augmented chord
figure = self.enc.decode_event(33)
self.assertEqual('Abaug', figure)
# diminished chord
figure = self.enc.decode_event(42)
self.assertEqual('Fdim', figure)
class PitchChordsEncoderDecoderTest(absltest.TestCase):
def setUp(self):
self.enc = chords_encoder_decoder.PitchChordsEncoderDecoder()
def testInputSize(self):
self.assertEqual(37, self.enc.input_size)
def testEncodeNoChord(self):
input_ = self.enc.events_to_input([NO_CHORD], 0)
self.assertEqual([1.0] + [0.0] * 36, input_)
def testEncodeChord(self):
# major triad
input_ = self.enc.events_to_input(['C'], 0)
expected = [0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
self.assertEqual(expected, input_)
# minor triad
input_ = self.enc.events_to_input(['F#m'], 0)
expected = [0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0]
self.assertEqual(expected, input_)
# major triad with dominant 7th in bass
input_ = self.enc.events_to_input(['G/F'], 0)
expected = [0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
self.assertEqual(expected, input_)
# 13th chord
input_ = self.enc.events_to_input(['E13'], 0)
expected = [0.0,
0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 1.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
self.assertEqual(expected, input_)
# minor triad with major 7th
input_ = self.enc.events_to_input(['Fm(maj7)'], 0)
expected = [0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
self.assertEqual(expected, input_)
if __name__ == '__main__':
absltest.main()Classes
class MajorMinorChordOneHotEncodingTest (*args, **kwargs)-
Extension of unittest.TestCase providing more power.
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 MajorMinorChordOneHotEncodingTest(absltest.TestCase): def setUp(self): self.enc = chords_encoder_decoder.MajorMinorChordOneHotEncoding() def testEncodeNoChord(self): index = self.enc.encode_event(NO_CHORD) self.assertEqual(0, index) def testEncodeChord(self): # major triad index = self.enc.encode_event('C') self.assertEqual(1, index) # minor triad index = self.enc.encode_event('Cm') self.assertEqual(13, index) # dominant 7th index = self.enc.encode_event('F7') self.assertEqual(6, index) # minor 9th index = self.enc.encode_event('Abm9') self.assertEqual(21, index) def testEncodeThirdlessChord(self): # suspended chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Gsus4') # power chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Bb5') def testDecodeNoChord(self): figure = self.enc.decode_event(0) self.assertEqual(NO_CHORD, figure) def testDecodeChord(self): # major chord figure = self.enc.decode_event(3) self.assertEqual('D', figure) # minor chord figure = self.enc.decode_event(17) self.assertEqual('Em', figure)Ancestors
- absl.testing.absltest.TestCase
- absl.third_party.unittest3_backport.case.TestCase
- unittest.case.TestCase
Methods
def setUp(self)-
Hook method for setting up the test fixture before exercising it.
Expand source code
def setUp(self): self.enc = chords_encoder_decoder.MajorMinorChordOneHotEncoding() def testDecodeChord(self)-
Expand source code
def testDecodeChord(self): # major chord figure = self.enc.decode_event(3) self.assertEqual('D', figure) # minor chord figure = self.enc.decode_event(17) self.assertEqual('Em', figure) def testDecodeNoChord(self)-
Expand source code
def testDecodeNoChord(self): figure = self.enc.decode_event(0) self.assertEqual(NO_CHORD, figure) def testEncodeChord(self)-
Expand source code
def testEncodeChord(self): # major triad index = self.enc.encode_event('C') self.assertEqual(1, index) # minor triad index = self.enc.encode_event('Cm') self.assertEqual(13, index) # dominant 7th index = self.enc.encode_event('F7') self.assertEqual(6, index) # minor 9th index = self.enc.encode_event('Abm9') self.assertEqual(21, index) def testEncodeNoChord(self)-
Expand source code
def testEncodeNoChord(self): index = self.enc.encode_event(NO_CHORD) self.assertEqual(0, index) def testEncodeThirdlessChord(self)-
Expand source code
def testEncodeThirdlessChord(self): # suspended chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Gsus4') # power chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Bb5')
class PitchChordsEncoderDecoderTest (*args, **kwargs)-
Extension of unittest.TestCase providing more power.
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 PitchChordsEncoderDecoderTest(absltest.TestCase): def setUp(self): self.enc = chords_encoder_decoder.PitchChordsEncoderDecoder() def testInputSize(self): self.assertEqual(37, self.enc.input_size) def testEncodeNoChord(self): input_ = self.enc.events_to_input([NO_CHORD], 0) self.assertEqual([1.0] + [0.0] * 36, input_) def testEncodeChord(self): # major triad input_ = self.enc.events_to_input(['C'], 0) expected = [0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # minor triad input_ = self.enc.events_to_input(['F#m'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # major triad with dominant 7th in bass input_ = self.enc.events_to_input(['G/F'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # 13th chord input_ = self.enc.events_to_input(['E13'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # minor triad with major 7th input_ = self.enc.events_to_input(['Fm(maj7)'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_)Ancestors
- absl.testing.absltest.TestCase
- absl.third_party.unittest3_backport.case.TestCase
- unittest.case.TestCase
Methods
def setUp(self)-
Hook method for setting up the test fixture before exercising it.
Expand source code
def setUp(self): self.enc = chords_encoder_decoder.PitchChordsEncoderDecoder() def testEncodeChord(self)-
Expand source code
def testEncodeChord(self): # major triad input_ = self.enc.events_to_input(['C'], 0) expected = [0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # minor triad input_ = self.enc.events_to_input(['F#m'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # major triad with dominant 7th in bass input_ = self.enc.events_to_input(['G/F'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # 13th chord input_ = self.enc.events_to_input(['E13'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) # minor triad with major 7th input_ = self.enc.events_to_input(['Fm(maj7)'], 0) expected = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] self.assertEqual(expected, input_) def testEncodeNoChord(self)-
Expand source code
def testEncodeNoChord(self): input_ = self.enc.events_to_input([NO_CHORD], 0) self.assertEqual([1.0] + [0.0] * 36, input_) def testInputSize(self)-
Expand source code
def testInputSize(self): self.assertEqual(37, self.enc.input_size)
class TriadChordOneHotEncodingTest (*args, **kwargs)-
Extension of unittest.TestCase providing more power.
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 TriadChordOneHotEncodingTest(absltest.TestCase): def setUp(self): self.enc = chords_encoder_decoder.TriadChordOneHotEncoding() def testEncodeNoChord(self): index = self.enc.encode_event(NO_CHORD) self.assertEqual(0, index) def testEncodeChord(self): # major triad index = self.enc.encode_event('C13') self.assertEqual(1, index) # minor triad index = self.enc.encode_event('Cm(maj7)') self.assertEqual(13, index) # augmented triad index = self.enc.encode_event('Faug7') self.assertEqual(30, index) # diminished triad index = self.enc.encode_event('Abm7b5') self.assertEqual(45, index) def testEncodeThirdlessChord(self): # suspended chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Gsus4') # power chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Bb5') def testDecodeNoChord(self): figure = self.enc.decode_event(0) self.assertEqual(NO_CHORD, figure) def testDecodeChord(self): # major chord figure = self.enc.decode_event(3) self.assertEqual('D', figure) # minor chord figure = self.enc.decode_event(17) self.assertEqual('Em', figure) # augmented chord figure = self.enc.decode_event(33) self.assertEqual('Abaug', figure) # diminished chord figure = self.enc.decode_event(42) self.assertEqual('Fdim', figure)Ancestors
- absl.testing.absltest.TestCase
- absl.third_party.unittest3_backport.case.TestCase
- unittest.case.TestCase
Methods
def setUp(self)-
Hook method for setting up the test fixture before exercising it.
Expand source code
def setUp(self): self.enc = chords_encoder_decoder.TriadChordOneHotEncoding() def testDecodeChord(self)-
Expand source code
def testDecodeChord(self): # major chord figure = self.enc.decode_event(3) self.assertEqual('D', figure) # minor chord figure = self.enc.decode_event(17) self.assertEqual('Em', figure) # augmented chord figure = self.enc.decode_event(33) self.assertEqual('Abaug', figure) # diminished chord figure = self.enc.decode_event(42) self.assertEqual('Fdim', figure) def testDecodeNoChord(self)-
Expand source code
def testDecodeNoChord(self): figure = self.enc.decode_event(0) self.assertEqual(NO_CHORD, figure) def testEncodeChord(self)-
Expand source code
def testEncodeChord(self): # major triad index = self.enc.encode_event('C13') self.assertEqual(1, index) # minor triad index = self.enc.encode_event('Cm(maj7)') self.assertEqual(13, index) # augmented triad index = self.enc.encode_event('Faug7') self.assertEqual(30, index) # diminished triad index = self.enc.encode_event('Abm7b5') self.assertEqual(45, index) def testEncodeNoChord(self)-
Expand source code
def testEncodeNoChord(self): index = self.enc.encode_event(NO_CHORD) self.assertEqual(0, index) def testEncodeThirdlessChord(self)-
Expand source code
def testEncodeThirdlessChord(self): # suspended chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Gsus4') # power chord with self.assertRaises(chords_encoder_decoder.ChordEncodingError): self.enc.encode_event('Bb5')