def test_album_quantity_is_medium(self):
acdc = Artist("AC/DC")
album = Album("Back in Black", acdc, 7)
self.assertEqual("medium", album.stock_level())
def test_album_quantity_is_low(self):
acdc = Artist("AC/DC")
album = Album("Back in Black", acdc, 4)
self.assertEqual("low", album.stock_level())
def test_album_quantity_is_high(self):
acdc = Artist("AC/DC")
album = Album("Back in Black", acdc, 10)
self.assertEqual("high", album.stock_level())