class TwitterTest(unittest.TestCase): def setUp(self): self.scraper = Twitter() def test_connection(self): self.assertEqual(self.scraper.connection(), True) def test_grab_user(self): self.scraper.connection() self.assertEqual(self.scraper.grab_user(profile='ribeiroit'), True) self.assertEqual(self.scraper.profile['nm'], 'Thiago Ribeiro')