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))
