# test_songs.py


from songs import Song, SongCollection
import unittest



class SongCollectionTest(unittest.TestCase):
    def test_new_collections_have_size_zero(self):
        collection = SongCollection()
        self.assertEqual(collection.size(), 0)


    def test_after_adding_one_song_to_a_collection__size_is_1(self):
        collection = SongCollection()
        collection.add(Song())
        self.assertEqual(collection.size(), 1)



if __name__ == '__main__':
    unittest.main()