class FragmentTest(TestCase): def setUp(self): self.urlValidator = UrlValidator() # input: correct input # expected: non-None def test_basicCorrectSituation(self): urls = ['#fragment',\ '#_-_',\ '#'] expected = ['',\ '',\ ''] for i in range(0, len(expected)): ret = self.urlValidator._isFragmentValid(urls[i]) self.assertEqual(expected[i], ret, 'wrong val at url: ' + urls[i]) # input: incorrect input # expected: None def test_illegalPath(self): urls = ['##',\ '#omg this is a fragment?', \ '#www.google.com'] for url in urls: ret = self.urlValidator._isFragmentValid(url) self.assertEqual(None, ret, 'wrong val at url: ' + url) # input: no Fragment # expected: original url def test_noFragment(self): urls = [''] for url in urls: ret = self.urlValidator._isFragmentValid(url) self.assertEqual(url, ret, 'wrong val at url: ' + url)