class TestBarChords(unittest.TestCase): def setUp(self): song_data = self.__song_data = SongData([ BarInfo() ], [], 0) self.__bar_chords1 = BarChords() self.__bar_chords1.set_song_data(song_data) self.__bar_chords1.set_chords([['CM7', ' trail1 '], [ 'Am', ' trail2 ']]) self.__bar_chords2 = BarChords() self.__bar_chords2.set_song_data(song_data) self.__bar_chords2.set_chords([['CM7', ' trail1 '], [ 'Am', ' trail2 '], [ '/', ' '], ['G', ' trail3 ']]) def test_set_chord1(self): bar_chords = self.__bar_chords1 bar_chords.set_chord(0, 'A') assert bar_chords.get_chords() == [['A', ' trail1 '], [ 'Am', ' trail2 ']] bar_chords.set_chord(0, '') assert bar_chords.get_chords() == [['/', ' trail1 '], [ 'Am', ' trail2 ']] # remove the last chord assert '\n' == bar_chords.get_eol() bar_chords.set_chord(1, '') assert bar_chords.get_chords() == [['/', ' trail1 ']] assert ' trail2 \n' == bar_chords.get_eol() bar_chords.set_chord(0, '') assert bar_chords.get_chords() == [['/', ' trail1 ']] bar_chords.set_chord(3, 'BM6') assert bar_chords.get_chords() == [['/', ' trail1 '], ['/', ' '], ['/', ' '], ['BM6', '']] def test_set_chord2(self): bar_chords = self.__bar_chords2 bar_chords.set_chord(2, 'B') assert bar_chords.get_chords() == [['CM7', ' trail1 '], [ 'Am', ' trail2 '], [ 'B', ' '], ['G', ' trail3 ']] bar_chords.set_chord(0, '') assert bar_chords.get_chords() == [['/', ' trail1 '], [ 'Am', ' trail2 '], [ 'B', ' '], ['G', ' trail3 ']] # remove the last chord assert '\n' == bar_chords.get_eol() bar_chords.set_chord(3, '') assert bar_chords.get_chords() == [['/', ' trail1 '], [ 'Am', ' trail2 '], [ 'B', ' ']] assert ' trail3 \n' == bar_chords.get_eol()
def create_bar_chords(self): bar_chords = BarChords() bar_chords.set_song_data(self) return bar_chords