def testCreateFPLink(self): self.assertEqual('http://www.fantasypros.com/nfl/rankings/ros-qb.php?export=xls', Ranking.create_standard_link('QB')) self.assertEqual('http://www.fantasypros.com/nfl/rankings/ros-ppr-te.php?export=xls', Ranking.create_ppr_link('TE')) self.assertEqual([('standard', 'http://www.fantasypros.com/nfl/rankings/ros-' 'wr.php?export=xls'), ('half_ppr', 'http://www.fantasypros.com/nfl/rankings/ros-' 'half-point-ppr-wr.php?export=xls'), ('ppr', 'http://www.fantasypros.com/nfl/rankings/ros-' 'ppr-wr.php?export=xls')], Ranking.create_all_links('WR'))
def testDownloadRankings(self, urllib_mock): urllib_mock.return_value.read.return_value = TestData('rbs.tsv') link = Ranking.create_ppr_link('RB') rankings = Ranking.download_rankings(link).splitlines() self.assertEqual('FantasyPros.com', rankings[1].strip()) self.assertEqual('Week 3 - RB Rankings', rankings[2].strip())