def test_zip(input, expected): assert s.zipcode(input) == expected
def zipcode(zip): return s.zipcode(zip)