class Test(unittest.TestCase):
def setUp(self):
self.album = Album()
def test_sample_1(self):
test_songs = [(30, "one"), (30, "two"), (15, "three"), (25, "four")]
self.album.load([Song(song) for song in test_songs])
self.assertEqual(self.album.find_bests(2), ["four", "two"])
def test_load_songs_from_file(self):
self.album.load_from_file("sample.txt")
