def test_set_attr(self):
attr = {"fire": True}
a = Track(track_id="123",
name="Trust Nobody",
album="Cheap Queen",
album_type="album")
a.set_attrs(attr)
assert a.attr == attr
def test_not_equal_id(self):
a = Track(track_id="123",
name="Trust Nobody",
album="Cheap Queen",
album_type="album")
b = Track(track_id="456",
name="Trust Nobody",
album="Cheap Queen",
album_type="album")
assert a != b
def test_equal_with_attr(self):
a = Track(track_id="123",
name="Trust Nobody",
album="Cheap Queen",
album_type="album",
attr={"fire": True})
b = Track(track_id="123",
name="Trust Nobody",
album="Cheap Queen",
album_type="album")
assert a == b
from acquisition.playlist import Playlist
from acquisition.track import Track
from acquisition.artist import Artist
playlist = Playlist(playlist_id="123",
name="sleep with a baseball bat",
entries=[(Track(track_id="000",
name="Sleep with a Baseball Bat",
album="Good Grief",
album_type="album"),
[Artist(artist_id="0001",
name="Cosmic Johnny")]),
(Track(track_id="111",
name="Riot for Descendant Command",
album="Oversleepers International",
album_type="album"),
[Artist(artist_id="1111", name="Emperor X")]),
(Track(track_id="222",
name="Forever",
album="Telefone",
album_type="album"), [
Artist(artist_id="2221",
name="Noname"),
Artist(artist_id="2222",
name="Joseph Chilliams"),
Artist(artist_id="2223",
name="Ravyn Lenae")
])])
class TestPlaylist:
import random
from pandas import DataFrame
from acquisition.graph import Graph
from acquisition.track import Track
from acquisition.artist import Artist
tracks = [(Track(track_id="000",
name="Diddy Bop",
album="Telefone",
album_type="album"), [
Artist(artist_id="0001", name="Noname"),
Artist(artist_id="0002", name="Cam O'bi"),
Artist(artist_id="0003", name="Raury")
]),
(Track(track_id="111",
name="Yesterday",
album="Telefone",
album_type="album"),
[Artist(artist_id="0001", name="Noname")]),
(Track(track_id="222",
name="Forever",
album="Telefone",
album_type="album"), [
Artist(artist_id="0001", name="Noname"),
Artist(artist_id="2222", name="Joseph Chilliams"),
Artist(artist_id="2223", name="Ravyn Lenae")
])]