Fixed audio_features API wrapper function return structure (#653)

Co-authored-by: Stéphane Bruckert <stephane.bruckert@gmail.com>
This commit is contained in:
badfarmerjohn 2021-03-05 01:36:07 -08:00 committed by dieser-niko
parent a614748ad7
commit 7bb3e7349f

View File

@ -87,8 +87,8 @@ class AuthTestSpotipy(unittest.TestCase):
bad_tracks = ['spotify:track:bad'] bad_tracks = ['spotify:track:bad']
input = self.four_tracks + bad_tracks input = self.four_tracks + bad_tracks
results = self.spotify.audio_features(input) results = self.spotify.audio_features(input)
self.assertTrue(len(results) == len(input)) self.assertTrue(len(results['audio_features']) == len(input))
for track in results[:-1]: for track in results['audio_features'][:-1]:
if track is not None: if track is not None:
assert ('speechiness' in track) assert ('speechiness' in track)
self.assertTrue(results[-1] is None) self.assertTrue(results[-1] is None)