def test_distance(self):
# Coords according to https://tools.wmflabs.org/geohack/
# Minneapolis
lat1 = 44.98
lng1 = -93.27
# Chicago
lat2 = 41.84
lng2 = -87.68
# Sao Paulo
lat3 = -23.55
lng3 = -46.63
# distances from http://www.movable-type.co.uk/scripts/latlong.html
# (Rounds to nearest km)
self.assertTrue(np.allclose(utils.distance(lat1, lng1, lat2, lng2),
570.6, atol=1, rtol=0))
self.assertTrue(np.allclose(utils.distance(lat1, lng1, lat3, lng3),
8966, atol=1, rtol=0))
def distance_from_point(lat_origin, lng_origin, lat_col, lng_col):
"""Generates a column of how far each record is from the origin
Parameters
----------
lat_origin : number
lng_origin : number
lat_col : np.ndarray
lng_col : np.ndarray
Returns
-------
np.ndarray
"""
lat_col = utils.check_col(lat_col, argument_name='lat_col')
lng_col = utils.check_col(lng_col, argument_name='lng_col')
return distance(lat_origin, lng_origin, lat_col, lng_col)
def distance_from_point(lat_origin, lng_origin, lat_col, lng_col):
"""Generates a column of how far each record is from the origin
Parameters
----------
lat_origin : number
lng_origin : number
lat_col : np.ndarray
lng_col : np.ndarray
Returns
-------
np.ndarray
"""
lat_col = utils.check_col(lat_col, argument_name='lat_col')
lng_col = utils.check_col(lng_col, argument_name='lng_col')
return distance(lat_origin, lng_origin, lat_col, lng_col)
