def test_number_starred_song(self): app = MusicApp(True) for k in range(20): app.add_song(Song(starred = not k % 2)) self.assertEqual(len(app.get_starred_songs()), 10)
def test_starred_song(self): app = MusicApp(True) for k in range(5): app.add_song(Song()) only_song_starred = Song(starred = True) app.add_song(only_song_starred) self.assertEqual(app.get_starred_songs()[0], only_song_starred)