示例#1
0
 def test_single(self):
     geotrie = GeoTrie(precision=6)
     point = (51.528642, -0.101599)
     geotrie.add(point, data='foo')
     assert geotrie.trie.has('gcpvjs')
     assert geotrie.trie.values('gcpvjs') == [
         ((51.528642, -0.101599), 'foo')
     ]
示例#2
0
    def test_search(self, precision, search):
        points, expected_hits = search

        geotrie = GeoTrie(precision=precision)
        for p in points:
            geotrie.add(p)

        center = (51.513284, -0.136539)
        hits = geotrie.radius_search(center, 500)
        assert len(hits) == len(expected_hits)

        hits = map(operator.itemgetter(0), hits)
        for i in expected_hits:
            assert points[i] in hits
示例#3
0
    def test_multiple(self):
        geotrie = GeoTrie(precision=6)

        point = (51.528642, -0.101599)
        geotrie.add(point)
        assert geotrie.trie.has('gcpvjs')
        assert geotrie.trie.values('gcpvjs') == [
            ((51.528642, -0.101599), None)
        ]

        point = (51.7505017, -1.3177993)
        geotrie.add(point)
        assert geotrie.trie.has('gcpn6h')
        assert geotrie.trie.values('gcpn6h') == [
            ((51.7505017, -1.3177993), None)
        ]