Coverage for src/aggregator/classes/album.py: 100%

1"""Module for album class.""" 

2 

3 

4class Album: 

5 """Class representing an album.""" 

6 def __init__(self, album_name, album_id, album_type, release_date, genres, label, popularity): 

7 self.album_name = album_name 

8 self.album_id = album_id 

9 self.album_type = album_type 

10 self.release_date = release_date 

11 self.genres = genres 

12 self.label = label 

13 self.popularity = popularity 

14 if type(album_name) is not str: 

15 raise TypeError("Album name must be string.") 

16 

17 def __str__(self): 

18 return f'{self.album_name} {self.album_id} {self.album_type}' \ 

19 f' {self.release_date} {self.genres} {self.label} {self.popularity}' 

20 

21 def get_album_name(self): 

22 """Function returns album_name.""" 

23 return self.album_name