29 lines
833 B
Python
29 lines
833 B
Python
|
"""Tests to converting MusicXML to NoteSequence proto."""
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
# internal imports
|
||
|
|
import tensorflow as tf
|
||
|
|
|
||
|
|
from magenta.music.music_xml_io import music_xml_to_sequence_proto
|
||
|
|
|
||
|
|
|
||
|
|
class MusicXmlIOTest(tf.test.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Get the file path to the test MusicXML file."""
|
||
|
|
fname = 'bach-one_phrase-4_voices.xml'
|
||
|
|
self.music_xml_fpath = os.path.join(
|
||
|
|
tf.resource_loader.get_data_files_path(), 'testdata', fname)
|
||
|
|
|
||
|
|
def testMusicXMLToNoteSequence(self):
|
||
|
|
"""A few quick tests for converting MusicXML to NoteSequence proto."""
|
||
|
|
sequence = music_xml_to_sequence_proto(self.music_xml_fpath)
|
||
|
|
self.assertEqual(len(sequence.time_signatures), 2)
|
||
|
|
self.assertEqual(len(sequence.key_signatures), 1)
|
||
|
|
self.assertEqual(len(sequence.notes), 47)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
tf.test.main()
|