Skip to content

Commit

Permalink
rtree.index: fix test bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
adamjstewart committed Jan 6, 2025
1 parent 3512cac commit f8a88aa
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tests/test_index.py
Original file line number Diff line number Diff line change
Expand Up @@ -240,7 +240,7 @@ def test_intersection(self) -> None:
self.assertTrue(0 in self.idx.intersection((0, 0, 60, 60)))
hits = list(self.idx.intersection((0, 0, 60, 60)))

self.assertTrue(len(hits), 10)
self.assertEqual(len(hits), 10)
self.assertEqual(hits, [0, 4, 16, 27, 35, 40, 47, 50, 76, 80])

def test_objects(self) -> None:
Expand Down Expand Up @@ -436,14 +436,14 @@ def test_custom_filenames(self) -> None:
idx.add(i, coords)

hits = list(idx.intersection((0, 0, 60, 60)))
self.assertTrue(len(hits), 10)
self.assertEqual(len(hits), 10)
self.assertEqual(hits, [0, 4, 16, 27, 35, 40, 47, 50, 76, 80])
del idx

# Check we can reopen the index and get the same results
idx2 = index.Index(tname, properties=p)
hits = list(idx2.intersection((0, 0, 60, 60)))
self.assertTrue(len(hits), 10)
self.assertEqual(len(hits), 10)
self.assertEqual(hits, [0, 4, 16, 27, 35, 40, 47, 50, 76, 80])

@pytest.mark.skipif(not sys.maxsize > 2**32, reason="Fails on 32bit systems")
Expand All @@ -465,7 +465,7 @@ def data_gen(
tname, data_gen(interleaved=False), properties=p, interleaved=False
)
hits1 = sorted(list(idx.intersection((0, 60, 0, 60))))
self.assertTrue(len(hits1), 10)
self.assertEqual(len(hits1), 10)
self.assertEqual(hits1, [0, 4, 16, 27, 35, 40, 47, 50, 76, 80])

leaves = idx.leaves()
Expand Down Expand Up @@ -591,7 +591,7 @@ def data_gen(
)

hits2 = sorted(list(idx.intersection((0, 60, 0, 60), objects=True)))
self.assertTrue(len(hits2), 10)
self.assertEqual(len(hits2), 10)
self.assertEqual(hits2[0].object, 42)

def test_overwrite(self) -> None:
Expand Down Expand Up @@ -855,6 +855,6 @@ def test_custom_storage_reopening(self) -> None:
del r1
self.assertTrue(storage.hasData)

r2 = index.Index(storage, properly=settings, overwrite=False)
r2 = index.Index(storage, properties=settings, overwrite=False)
count = r2.count((0, 0, 10, 10))
self.assertEqual(count, 1)

0 comments on commit f8a88aa

Please sign in to comment.