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)