Skip to content

Commit 485f084

Browse files
committed
fix: adding test to bb
1 parent 0ff6735 commit 485f084

1 file changed

Lines changed: 58 additions & 0 deletions

File tree

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
from LoopStructural.utils import BoundingBox
2+
import numpy as np
3+
4+
5+
def test_create_bounding_box():
6+
bbox = BoundingBox(origin=[0, 0, 0], maximum=[1, 1, 1])
7+
assert bbox.origin[0] == 0
8+
assert bbox.origin[1] == 0
9+
assert bbox.origin[2] == 0
10+
assert bbox.maximum[0] == 1
11+
assert bbox.maximum[1] == 1
12+
assert bbox.maximum[2] == 1
13+
assert np.all(bbox.bb == np.array([[0, 0, 0], [1, 1, 1]]))
14+
assert bbox.valid == True
15+
16+
17+
def test_create_bounding_box_from_points():
18+
bbox = BoundingBox()
19+
bbox.fit(np.array([[0, 0, 0], [1, 1, 1]]))
20+
assert bbox.origin[0] == 0
21+
assert bbox.origin[1] == 0
22+
assert bbox.origin[2] == 0
23+
assert bbox.maximum[0] == 1
24+
assert bbox.maximum[1] == 1
25+
assert bbox.maximum[2] == 1
26+
assert np.all(bbox.bb == np.array([[0, 0, 0], [1, 1, 1]]))
27+
assert bbox.valid == True
28+
29+
30+
def test_create_with_buffer():
31+
bbox = BoundingBox(origin=[0, 0, 0], maximum=[1, 1, 1])
32+
bbox = bbox.with_buffer(0.2)
33+
assert bbox.origin[0] == -0.2
34+
assert bbox.origin[1] == -0.2
35+
assert bbox.origin[2] == -0.2
36+
assert bbox.maximum[0] == 1.2
37+
assert bbox.maximum[1] == 1.2
38+
assert bbox.maximum[2] == 1.2
39+
assert np.all(bbox.bb == np.array([[-0.2, -0.2, -0.2], [1.2, 1.2, 1.2]]))
40+
assert bbox.valid == True
41+
42+
43+
def test_is_inside():
44+
bbox = BoundingBox(origin=[0, 0, 0], maximum=[1, 1, 1])
45+
assert bbox.is_inside(np.array([0.5, 0.5, 0.5])) == True
46+
assert bbox.is_inside(np.array([0.5, 0.5, 1.5])) == False
47+
assert bbox.is_inside(np.array([0.5, 0.5, -0.5])) == False
48+
assert bbox.is_inside(np.array([0.5, 1.5, 0.5])) == False
49+
assert bbox.is_inside(np.array([0.5, -0.5, 0.5])) == False
50+
assert bbox.is_inside(np.array([1.5, 0.5, 0.5])) == False
51+
assert bbox.is_inside(np.array([-0.5, 0.5, 0.5])) == False
52+
53+
54+
if __name__ == "__main__":
55+
test_create_bounding_box()
56+
test_create_bounding_box_from_points()
57+
test_create_with_buffer()
58+
test_is_inside()

0 commit comments

Comments
 (0)