def test_haversine_na(self):
km_dist_na = compareLatLong((0.0, 0.0), (1.0, 2.0))
assert numpy.isnan(km_dist_na)
km_dist_na = compareLatLong((1.0, 2.0), (0.0, 0.0))
assert numpy.isnan(km_dist_na)
km_dist_n_na = compareLatLong((0.0, 1.0), (1.0, 2.0))
self.assertAlmostEqual(km_dist_n_na, 157, -1)
def test_haversine_na(self):
km_dist_na = compareLatLong((0.0, 0.0), (1.0, 2.0))
assert numpy.isnan(km_dist_na)
km_dist_na = compareLatLong((1.0, 2.0), (0.0, 0.0))
assert numpy.isnan(km_dist_na)
km_dist_n_na = compareLatLong((0.0, 1.0), (1.0, 2.0))
self.assertAlmostEqual(km_dist_n_na, 157, -1)
df_patstat = pd.read_csv(patstat_file, sep='\t', usecols = ['person_id', 'lat', 'lng'])
df_patstat = df_patstat[pd.notnull(df_patstat['lat'])]
geodist = df_out[['patstat_id','company_id']].set_index('patstat_id').join(df_patstat.set_index('person_id'))
geodist = geodist.rename(columns = {'lat':'plat', 'lng': 'plng'})
geodist = geodist.reset_index().set_index('company_id').join(df_amadeus.set_index('company_id')).reset_index()
geodist = geodist.rename(columns = {'index':'patstat_id','lat':'alat', 'lng': 'alng'})
geodist = geodist[pd.notnull(geodist['plat']*geodist['alat'])]
# geocode matches
print 'computing geo-distances of selected matches'
geo_dist = []
for i,v in geodist.iterrows():
geo_dist.append(haversine.compareLatLong((v['plat'],v['plng']),(v['alat'],v['alng'])))
geodist['geo_dist'] = geo_dist
geodist = geodist[['patstat_id','company_id','geo_dist']].set_index(['patstat_id','company_id'])
df_out = df_out.set_index(['patstat_id','company_id']).join(geodist)
print 'done with geo-distances'
df_out['country'] = country
# save
print 'saving to file', output_file
df_out.to_csv(output_file, sep='\t')
print 'done!'
def test_haversine_zero(self):
km_dist_zero = compareLatLong(self.ord, self.ord)
self.assertAlmostEqual(km_dist_zero, 0.0, 0)
def test_haversine_equal(self):
km_dist_val = compareLatLong(self.sfo, self.ord)
self.assertAlmostEqual(km_dist_val, 2964, -1)
def test_haversine_equal(self):
km_dist_val = compareLatLong(self.sfo, self.ord)
self.assertAlmostEqual(km_dist_val, 2964, -1)
def test_haversine_zero(self):
km_dist_zero = compareLatLong(self.ord, self.ord)
self.assertAlmostEqual(km_dist_zero, 0.0, 0)
