def test_get_music_taste_with_audio_features_exists(self): """ If _music_taste has audio features in its data, music_taste_with_audio_features() should return it. """ u = UserData(None) test_data = [{'id': 5, 'energy': 0}, {'id': 2, 'energy': 5}] u._music_taste = test_data self.assertEqual(u.music_taste_with_audio_features(), test_data)
def test_compile_audio_features(self): """ _compile_audio_features() should request data about audio features of the user's music taste from Spotify and store it in the _music_taste variable. """ u = UserData(self.session) data = u.music_taste_with_audio_features() self.assertGreaterEqual(len(data), 50) for t in data: self.assertEqual(t['type'], 'track') self.assertIsNotNone(t.get('energy')) self.assertIsNotNone(t.get('danceability')) self.assertIsNotNone(t.get('valence'))