class TestSinkSa(unittest.TestCase): def setUp(self): self._sink = Sink() def test_get_titles_rating(self): count = 0 for t in self._sink.get_titles_rating(): count += 1 for k in ('title_id', 'imdb_rating', 'imdb_votes', 'netflix_rating'): self.assertTrue(k in t) self.assertFalse(k is None) self.assertTrue(count > 100)
def setUp(self): self._sink = Sink()