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)