def test_crawl_single(self): twit = MockTwitterResource() gis = MockGisgraphyResource() user = User(_id=6) results = sprawl.crawl_single(user, twit, gis, fast=False) self.assertEqual(results.nebrs[0]._id, 1) self.assertEqual(results.nebrs[0].local_ratio, 1) self.assertEqual(results.ats, [7])
def _crawl_pred_one(user, twit, gis, pred, fast): if user.location and not user.geonames_place: user.geonames_place = gis.twitter_loc(user.location) gnp = user.geonames_place if gnp and gnp.mdist < MAX_GNP_MDIST: user.pred_loc = gnp.to_tup() return nebrs, ats, ated = sprawl.crawl_single(user, twit, gis, fast=fast) if nebrs: clf = "friendloc_nearcut" if fast else "friendloc_cut" user.pred_loc = pred.predict(user, nebrs, ats, ated, clf)