def test_good(self):
self.code = ZipCode.check_zipcode("90231-9023")
correct_code = "90231-9023"
self.assertEqual( correct_code, self.code)
def test_middle(self):
self.code = ZipCode.check_zipcode("9-8 201")
correct_code = "98201"
self.assertEqual( correct_code, self.code)
def test_vbad(self):
self.code = ZipCode.check_zipcode("oyhgegs")
correct_code = ""
self.assertEqual( correct_code, self.code)
def test_bad(self):
self.code = ZipCode.check_zipcode("^89623y8o")
correct_code = ""
self.assertEqual( correct_code, self.code)
def setUp(self):
self.code = ZipCode.check_zipcode("")
