def test_multiple_cells(self): now = util.utcnow() cell = CellShardFactory(samples=100) cell2 = CellShardFactory(radio=cell.radio, mcc=cell.mcc, mnc=cell.mnc, lac=cell.lac, cid=cell.cid + 1, lat=cell.lat + 1.0, lon=cell.lon + 1.0, samples=10) self.session.flush() query = self.model_query(cells=[cell, cell2]) results = self.source.search(query) self.check_model_results( results, [cell], lat=cell.lat + 0.3333333, lon=cell.lon + 0.3333333, accuracy=CELL_MAX_ACCURACY) self.assertAlmostEqual( results.best().score, cell.score(now) + cell2.score(now), 4)
def test_cell(self): now = util.utcnow() cell = CellShardFactory(samples=10) self.session.flush() query = self.model_query(cells=[cell]) results = self.source.search(query) self.check_model_results(results, [cell]) self.assertAlmostEqual(results.best().score, cell.score(now), 4)
def test_cell(self, geoip_db, http_session, session, source, stats): now = util.utcnow() cell = CellShardFactory(samples=10) session.flush() query = self.model_query( geoip_db, http_session, session, stats, cells=[cell]) results = source.search(query) self.check_model_results(results, [cell]) assert results.best().score == cell.score(now)