class TestClustering(unittest.TestCase):
def setUp(self):
# Parse data dumps
self.search_term = "kanye west"
self.parser = ParseDataDumps()
self.parser.parse_metro_artist_chart( "test_artist_dump.json" )
self.parser.parse_top_tags( "tag_dump.json" )
self.artist_rankings = self.parser.artist_rankings
self.search_rankings = sorted(self.artist_rankings[self.search_term], key=lambda city:city[1])
self.artist_tags = self.parser.artist_tags[self.search_term]
#print artist_tags
def test_rank(self):
artist_rankings = self.parser.artist_rankings
search_rankings = sorted(self.artist_rankings[self.search_term], key=lambda city:city[1])
#self.artist_tags = self.parser.artist_tags[self.search_term]
#print artist_tags
self.assertEqual(self.search_rankings[0][1],1)
def test_top_tag(self):
artist_tags = self.parser.artist_tags[self.search_term]
print artist_tags
self.assertEqual(self.artist_tags[0][0], "Hip-Hop")
self.assertEqual(self.artist_tags[0][1], 100)
