def test_best_m_songs(self):
#Test m = 1 and m = n
songs = [
{'title': 'one', 'plays': '3', 'index': 1},
{'title': 'two', 'plays': '1', 'index': 2}, ]
self.assertEqual(zipfsong.best_m_songs(2, songs), songs)
self.assertEqual(zipfsong.best_m_songs(1, songs), songs[:1])
self.random_tests(100)
def random_tests(self, num_tests):
for i in range(num_tests):
#Pick random values for n and m
n = random.randint(1, 50000)
m = random.randint(1, n)
(songs, best_indices) = self.random_album(n, m)
best_m_songs = zipfsong.best_m_songs(m, songs)
self.assertEqual([song['index'] for song in best_m_songs], best_indices)
