def test_geohash_encode():
# encode lon/lat into geohash
post_df = geohash_encode(
df=lonlat_df[["city", "latitude", "longitude"]],
latitude="latitude",
longitude="longitude",
geohash="geohash",
)
assert sorted(post_df.columns.tolist()) == sorted(
["city", "geohash", "latitude", "longitude"])
assert series_to_list(post_df["geohash"]) == series_to_list(
lonlat_df["geohash"])
def test_geohash_encode(self):
# encode lon/lat into geohash
post_df = proc.geohash_encode(
df=lonlat_df[["city", "latitude", "longitude"]],
latitude="latitude",
longitude="longitude",
geohash="geohash",
)
self.assertListEqual(
sorted(post_df.columns.tolist()),
sorted(["city", "geohash", "latitude", "longitude"]),
)
self.assertListEqual(
series_to_list(post_df["geohash"]), series_to_list(lonlat_df["geohash"]),
)