Kevin-Mok/spotify-lib-vis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Make genre data test more comprehensive

Browse Source 
Modified the genre data retrieval test to be more comprehensive.
This commit is contained in:
Chris Shyi
2018-07-29 22:38:40 -04:00
parent 765141c024
commit cc071d621c
2 changed files with 45 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
api/tests.py
View File

@@ -7,7 +7,8 @@ import pprint
class GenreDataTestCase(TestCase):
def setUp(self):
@classmethod
def setUpTestData(cls):
test_user = User.objects.create(id="chrisshi", refresh_token="blah", access_token="blah",
access_expires_in=10)
genre = Genre.objects.create(name="classical", num_songs=3)
@@ -40,11 +41,49 @@ class GenreDataTestCase(TestCase):
track_3.artists.add(artist_1)
track_3.artists.add(artist_3)
def test_get_artist_counts_in_genre(self):
pop_genre = Genre.objects.create(name='pop', num_songs=3)
pop_artist1 = Artist.objects.create(id='art4', name="Taylor Swift")
pop_artist2 = Artist.objects.create(id='art5', name="Justin Bieber")
pop_artist3 = Artist.objects.create(id='art6', name="Rihanna")
pop_track_1 = Track.objects.create(id='track4', year=2013,
popularity=5, runtime=20,
name='poptrack1',
genre=pop_genre)
pop_track_1.users.add(test_user)
pop_track_1.artists.add(pop_artist1)
pop_track_1.artists.add(pop_artist2)
pop_track_2 = Track.objects.create(id='track5', year=2013,
popularity=5, runtime=20,
name='poptrack2',
genre=pop_genre)
pop_track_2.users.add(test_user)
pop_track_2.artists.add(pop_artist3)
pop_track_2.artists.add(pop_artist2)
pop_track_2.artists.add(pop_artist1)
pop_track_3 = Track.objects.create(id='track6', year=2013,
popularity=5, runtime=20,
name='poptrack3',
genre=pop_genre)
pop_track_3.users.add(test_user)
pop_track_3.artists.add(pop_artist3)
pop_track_3.artists.add(pop_artist2)
pop_track_3.artists.add(pop_artist1)
def test_get_artist_counts_two_genres(self):
test_user = User.objects.get(id='chrisshi')
artist_counts = utils.get_artists_in_genre(test_user, 'classical', 10)
artist_counts = utils.get_artists_in_genre(test_user, 'classical')
# pprint.pprint(artist_counts)
self.assertTrue(math.isclose(artist_counts['Beethoven'], 1.3, rel_tol=0.05))
self.assertTrue(math.isclose(artist_counts['Mozart'], 0.85, rel_tol=0.05))
self.assertTrue(math.isclose(artist_counts['Chopin'], 0.85, rel_tol=0.05))
self.assertTrue(math.isclose(sum(artist_counts.values()), 3, rel_tol=0.01))
# test the pop genre
artist_counts = utils.get_artists_in_genre(test_user, 'pop')
self.assertTrue(math.isclose(artist_counts['Taylor Swift'], 1.125, rel_tol=0.05))
self.assertTrue(math.isclose(artist_counts['Justin Bieber'], 1.125, rel_tol=0.05))
self.assertTrue(math.isclose(artist_counts['Rihanna'], 0.75, rel_tol=0.05))
self.assertTrue(math.isclose(sum(artist_counts.values()), 3, rel_tol=0.01))

9
api/views.py
View File

@@ -23,9 +23,9 @@ ARTIST_LIMIT = 50
FEATURES_LIMIT = 100
# ARTIST_LIMIT = 25
# FEATURES_LIMIT = 25
TRACKS_TO_QUERY = 100
TRACKS_TO_QUERY = 300
console_logging = True
CONSOLE_LOGGING = True
# parse_library {{{ #
@@ -53,9 +53,6 @@ def parse_library(request, user_secret):
headers=user_headers,
params=payload).json()['items']
if console_logging:
tracks_processed = 0
tracks_processed = 0
for track_dict in saved_tracks_response:
# add artists {{{ #
@@ -92,7 +89,7 @@ def parse_library(request, user_secret):
# }}} add audio features #
if console_logging:
if CONSOLE_LOGGING:
tracks_processed += 1
print("Added track #{}: {} - {}".format(
offset + tracks_processed,