def test_postal_unknown(self): unknown_keys = [ "Wisco", "New Y." ] for key in unknown_keys: self.assertEqual(key, postal(key))
def test_postal(self): """ postal is essentially a lookup table that returns the arg when the key doesn't exist, so testing doesn't need to be very extensive """ knowns_key_vals = [ ("Wisconsin", "WI"), ("Tenn.", "TN"), ("Ore.", "OR"), ("Ohio", "OH") ] for name, code in knowns_key_vals: self.assertEqual(code, postal(name))