def test_error(self): with self.assertRaises(SystemExit): ipfromwebpage.check_args(self.bad_url)
def test_good_return(self): result = ipfromwebpage.check_args(self.good_url.split()) self.assertEqual(result.url, self.good_url)
def test_no_arguments(self): with self.assertRaises(SystemExit): ipfromwebpage.check_args()