import pytest from yandex_music import BriefInfo @pytest.fixture(scope='class') def brief_info(artist, track, album, cover, playlist_id, video, chart, vinyl): return BriefInfo(artist, [album], [album], TestBriefInfo.last_release_ids, [track], [artist], [cover], TestBriefInfo.concerts, [video], [vinyl], TestBriefInfo.has_promotions, [playlist_id], [chart]) class TestBriefInfo: last_release_ids = [8501194, 8302547, 8302836, 8302450] concerts = None has_promotions = False def test_expected_values(self, brief_info, artist, track, album, cover, playlist_id, video, chart, vinyl): assert brief_info.artist == artist assert brief_info.albums == [album] assert brief_info.also_albums == [album] assert brief_info.last_release_ids == self.last_release_ids assert brief_info.popular_tracks == [track] assert brief_info.similar_artists == [artist] assert brief_info.all_covers == [cover] assert brief_info.concerts == self.concerts assert brief_info.videos == [video] assert brief_info.vinyls == [vinyl] assert brief_info.has_promotions == self.has_promotions assert brief_info.playlist_ids == [playlist_id] assert brief_info.tracks_in_chart == [chart] def test_de_json_required(self, client, artist, track, album, cover, playlist_id, video, vinyl): json_dict = {'artist': artist.to_dict(), 'albums': [album.to_dict()], 'also_albums': [album.to_dict()], 'last_release_ids': self.last_release_ids, 'popular_tracks': [track.to_dict()], 'similar_artists': [artist.to_dict()], 'all_covers': [cover.to_dict()], 'concerts': self.concerts, 'videos': [video.to_dict()], 'vinyls': [vinyl.to_dict()], 'has_promotions': self.has_promotions, 'playlist_ids': [playlist_id.to_dict()]} brief_info = BriefInfo.de_json(json_dict, client) assert brief_info.artist == artist assert brief_info.albums == [album] assert brief_info.also_albums == [album] assert brief_info.last_release_ids == self.last_release_ids assert brief_info.popular_tracks == [track] assert brief_info.similar_artists == [artist] assert brief_info.all_covers == [cover] assert brief_info.concerts == self.concerts assert brief_info.videos == [video] assert brief_info.vinyls == [vinyl] assert brief_info.has_promotions == self.has_promotions assert brief_info.playlist_ids == [playlist_id] def test_de_json_all(self, client, artist, track, album, cover, playlist_id, video, chart, vinyl): json_dict = {'artist': artist.to_dict(), 'albums': [album.to_dict()], 'also_albums': [album.to_dict()], 'last_release_ids': self.last_release_ids, 'popular_tracks': [track.to_dict()], 'similar_artists': [artist.to_dict()], 'all_covers': [cover.to_dict()], 'concerts': self.concerts, 'videos': [video.to_dict()], 'vinyls': [vinyl.to_dict()], 'has_promotions': self.has_promotions, 'playlist_ids': [playlist_id.to_dict()], 'tracks_in_chart': [chart.to_dict()]} brief_info = BriefInfo.de_json(json_dict, client) assert brief_info.artist == artist assert brief_info.albums == [album] assert brief_info.also_albums == [album] assert brief_info.last_release_ids == self.last_release_ids assert brief_info.popular_tracks == [track] assert brief_info.similar_artists == [artist] assert brief_info.all_covers == [cover] assert brief_info.concerts == self.concerts assert brief_info.videos == [video] assert brief_info.vinyls == [vinyl] assert brief_info.has_promotions == self.has_promotions assert brief_info.playlist_ids == [playlist_id] assert brief_info.tracks_in_chart == [chart] def test_equality(self, artist, track, album, cover, playlist_id, video, vinyl): a = BriefInfo(artist, [album], [album], self.last_release_ids, [track], [artist], [cover], self.concerts, [video], [vinyl], self.has_promotions, [playlist_id]) b = BriefInfo(artist, [album], [album], self.last_release_ids, [], [artist], [cover], self.concerts, [video], [vinyl], True, [playlist_id]) c = BriefInfo(artist, [album], [album], [1, 2, 3], [track], [artist], [], self.concerts, [video], [vinyl], self.has_promotions, [playlist_id]) d = BriefInfo(artist, [album], [album], self.last_release_ids, [track], [artist], [cover], self.concerts, [video], [vinyl], self.has_promotions, [playlist_id]) assert a != b != c assert hash(a) != hash(b) != hash(c) assert a is not b is not c assert a == d