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
