diff --git a/test/test_octtree.py b/test/test_octtree.py index 6f39bcb4e6abb0264605667ac7ea251a98dd399e..8875b7a3f14c73b6e5128c1da8af098d94f8f2d6 100644 --- a/test/test_octtree.py +++ b/test/test_octtree.py @@ -136,6 +136,8 @@ class TestOctTree(unittest.TestCase): theta = 0 d = datetime(2023, 3, 24, 12, 0) + dt = timedelta(days=10) + test_datetime = d + timedelta(hours=4) test_timedelta = timedelta(hours=5) ellipse = Ellipse( @@ -147,8 +149,16 @@ class TestOctTree(unittest.TestCase): ellipse.p2_lat, ) - d = datetime(2023, 3, 24, 12, 0) - dt = timedelta(days=10) + # TEST: Near Boundary Points + assert ellipse.contains(Record(13.49, 2.5, test_datetime)) + assert ellipse.contains(Record(11.51, 2.5, test_datetime)) + assert ellipse.contains(Record(12.5, 2.99, test_datetime)) + assert ellipse.contains(Record(12.5, 2.01, test_datetime)) + assert not ellipse.contains(Record(13.51, 2.5, test_datetime)) + assert not ellipse.contains(Record(11.49, 2.5, test_datetime)) + assert not ellipse.contains(Record(12.5, 3.01, test_datetime)) + assert not ellipse.contains(Record(12.5, 1.99, test_datetime)) + boundary = Rectangle(10, 4, d, 20, 8, dt) otree = OctTree(boundary, capacity=3) diff --git a/test/test_quadtree.py b/test/test_quadtree.py index d11787224e5ac868345578e0e6cc898b70b08b9a..815d7c23b44c9aa3950b016b5ec99a50fc51a9c1 100644 --- a/test/test_quadtree.py +++ b/test/test_quadtree.py @@ -109,6 +109,17 @@ class TestQuadTree(unittest.TestCase): ellipse.p2_lon, ellipse.p2_lat, ) + + # TEST: Near Boundary Points + assert ellipse.contains(Record(13.49, 2.5)) + assert ellipse.contains(Record(11.51, 2.5)) + assert ellipse.contains(Record(12.5, 2.99)) + assert ellipse.contains(Record(12.5, 2.01)) + assert not ellipse.contains(Record(13.51, 2.5)) + assert not ellipse.contains(Record(11.49, 2.5)) + assert not ellipse.contains(Record(12.5, 3.01)) + assert not ellipse.contains(Record(12.5, 1.99)) + boundary = Rectangle(10, 4, 20, 8) qtree = QuadTree(boundary, capacity=3) points: list[Record] = [