mirrors/PMTiles
1
0
Fork 0
mirror of https://github.com/protomaps/PMTiles.git synced 2026-02-04 10:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

python reader: fix leaf level querying

Browse Source
This commit is contained in:
Brandon Liu
2022-07-18 21:00:46 +08:00
parent 0a153bf127
commit de687baf61
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/pmtiles/reader.py
View File

@@ -72,6 +72,8 @@ class Reader:
else: else:
if len(self.header().leaves) > 0: if len(self.header().leaves) > 0:
level_diff = z - self._leaf_level() level_diff = z - self._leaf_level()
if level_diff < 0:
return None
leaf = ( leaf = (
self._leaf_level(), self._leaf_level(),
x // (1 << level_diff), x // (1 << level_diff),

5
python/test/test_reader.py
View File

@@ -7,11 +7,10 @@ from pmtiles.reader import Reader, MemorySource
class TestReader(unittest.TestCase): class TestReader(unittest.TestCase):
def test_roundtrip(self): def test_roundtrip(self):
buf = BytesIO() buf = BytesIO()
writer = Writer(buf, 7) writer = Writer(buf, 5)
writer.write_tile(1, 0, 0, b"0") writer.write_tile(1, 0, 0, b"0")
writer.write_tile(1, 0, 1, b"1") writer.write_tile(1, 0, 1, b"1")
writer.write_tile(1, 1, 0, b"2") writer.write_tile(1, 1, 0, b"2")
writer.write_tile(1, 1, 1, b"3")
writer.write_tile(2, 0, 0, b"4") writer.write_tile(2, 0, 0, b"4")
writer.write_tile(3, 0, 0, b"5") writer.write_tile(3, 0, 0, b"5")
writer.write_tile(2, 0, 1, b"6") writer.write_tile(2, 0, 1, b"6")
@@ -24,8 +23,8 @@ class TestReader(unittest.TestCase):
self.assertEqual(reader.get(1, 0, 0), b"0") self.assertEqual(reader.get(1, 0, 0), b"0")
self.assertEqual(reader.get(1, 0, 1), b"1") self.assertEqual(reader.get(1, 0, 1), b"1")
self.assertEqual(reader.get(1, 1, 0), b"2") self.assertEqual(reader.get(1, 1, 0), b"2")
self.assertEqual(reader.get(1, 1, 1), b"3")
self.assertEqual(reader.get(2, 0, 0), b"4") self.assertEqual(reader.get(2, 0, 0), b"4")
self.assertEqual(reader.get(3, 0, 0), b"5") self.assertEqual(reader.get(3, 0, 0), b"5")
self.assertEqual(reader.get(2, 0, 1), b"6") self.assertEqual(reader.get(2, 0, 1), b"6")
self.assertEqual(reader.get(3, 0, 2), b"7") self.assertEqual(reader.get(3, 0, 2), b"7")
self.assertEqual(reader.get(1, 1, 1), None)