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)
