class TestWrite(unittest.TestCase):
reg = CounterRegistry()
def setUp(self):
self.measure = Measure(16)
counter = self.reg.getCounterByName("16ths")
mc = MeasureCount()
mc.addSimpleBeats(counter, 4)
self.measure.setBeatCount(mc)
def get_output(self):
handle = StringIO()
indenter = fileUtils.Indenter(handle)
self.measure.write(indenter)
return handle.getvalue().splitlines()
def testWriteEmpty(self):
output = self.get_output()
self.assertEqual(output,
['START_BAR 16',
' COUNT_INFO_START',
' REPEAT_BEATS 4',
' BEAT_START',
' COUNT |^e+a|',
' BEAT_END',
' COUNT_INFO_END',
' BARLINE NORMAL_BAR,NO_BAR',
' BARLINE NORMAL_BAR,NO_BAR',
'END_BAR'])
def testWriteSimple(self):
self.measure.addNote(NotePosition(noteTime = 0, drumIndex = 0), "a")
self.measure.addNote(NotePosition(noteTime = 1, drumIndex = 1), "b")
self.measure.addNote(NotePosition(noteTime = 2, drumIndex = 0), "c")
self.measure.addNote(NotePosition(noteTime = 3, drumIndex = 1), "d")
self.measure.addNote(NotePosition(noteTime = 4, drumIndex = 0), "e")
self.measure.addNote(NotePosition(noteTime = 5, drumIndex = 1), "f")
self.measure.addNote(NotePosition(noteTime = 6, drumIndex = 0), "g")
self.measure.addNote(NotePosition(noteTime = 7, drumIndex = 1), "h")
self.measure.addNote(NotePosition(noteTime = 8, drumIndex = 0), "i")
self.measure.addNote(NotePosition(noteTime = 9, drumIndex = 1), "j")
self.measure.addNote(NotePosition(noteTime = 10, drumIndex = 0), "k")
self.measure.addNote(NotePosition(noteTime = 11, drumIndex = 1), "l")
self.measure.addNote(NotePosition(noteTime = 12, drumIndex = 0), "m")
self.measure.addNote(NotePosition(noteTime = 13, drumIndex = 1), "n")
self.measure.addNote(NotePosition(noteTime = 14, drumIndex = 0), "o")
self.measure.addNote(NotePosition(noteTime = 15, drumIndex = 1), "p")
output = self.get_output()
self.assertEqual(output,
['START_BAR 16',
' COUNT_INFO_START',
' REPEAT_BEATS 4',
' BEAT_START',
' COUNT |^e+a|',
' BEAT_END',
' COUNT_INFO_END',
' BARLINE NORMAL_BAR,NO_BAR',
' NOTE 0,0,a',
' NOTE 1,1,b',
' NOTE 2,0,c',
' NOTE 3,1,d',
' NOTE 4,0,e',
' NOTE 5,1,f',
' NOTE 6,0,g',
' NOTE 7,1,h',
' NOTE 8,0,i',
' NOTE 9,1,j',
' NOTE 10,0,k',
' NOTE 11,1,l',
' NOTE 12,0,m',
' NOTE 13,1,n',
' NOTE 14,0,o',
' NOTE 15,1,p',
' BARLINE NORMAL_BAR,NO_BAR',
'END_BAR'])
def testWriteDecorations(self):
self.measure.setLineBreak(True)
self.measure.setSectionEnd(True)
self.measure.setRepeatEnd(True)
self.measure.setRepeatStart(True)
self.measure.alternateText = "xxx"
self.measure.repeatCount = 10
output = self.get_output()
self.assertEqual(output,
['START_BAR 16',
' COUNT_INFO_START',
' REPEAT_BEATS 4',
' BEAT_START',
' COUNT |^e+a|',
' BEAT_END',
' COUNT_INFO_END',
' BARLINE NORMAL_BAR,REPEAT_START,NO_BAR',
(' BARLINE NORMAL_BAR,LINE_BREAK,NO_BAR,' +
'SECTION_END,REPEAT_END'),
' REPEAT_COUNT 10',
' ALTERNATE xxx',
'END_BAR'])
