def test_three_chords(self): chords = [{0: 6}, {1: 12, 2: 16}, {4: 10, 5: 10}] avgs = _avg_frets(chords) self.assertEqual(avgs[0], 6) self.assertEqual(avgs[1], 14) self.assertEqual(avgs[2], 10)
def test_one_chord(self): chords = [{0: 5, 1: 9}] avgs = _avg_frets(chords) self.assertEqual(avgs[0], 0) self.assertEqual(avgs[1], 0) self.assertEqual(avgs[2], 7)