def test_in_between(self):
p1 = 284.651885, 130.233454, 553.35
p2 = 284.651885, 130.233454, 553.425
p3 = 284.651885, 130.233454, 553.5
p4 = 284.651885, 130.233454, 554.5
n1 = Node(p1, 1)
n2 = Node(p2, 2)
n3 = Node(p3, 3)
n4 = Node(p4, 4)
nodes = Nodes([n1, n2, n3, n4])
res = Nodes(nodes.get_by_volume(p=p1))
self.assertEqual(len(res), 1)
def test_get_by_volume_box(self):
n1, n2, n3, n4, n5, n6, n7, n8, n9, n10 = get_nodes()
s = Nodes(get_nodes())
c = Nodes(s.get_by_volume(p=(1.5, 0.5, 0.5), vol_box=(4.5, 5.5, 8.5)))
self.assertEqual(c, Nodes([n3, n7, n8]))
def test_get_by_volume_cylinder(self):
n1, n2, n3, n4, n5, n6, n7, n8, n9, n10 = get_nodes()
s = Nodes(get_nodes())
c = Nodes(s.get_by_volume(p=(1.0, 1.0, 0.5), vol_cyl=(0.2, 4, 0.2)))
self.assertEqual(c, Nodes([n2, n6, n9]))
def test_get_by_volume_point(self):
n1, n2, n3, n4, n5, n6, n7, n8, n9, n10 = get_nodes()
s = Nodes(get_nodes())
c = Nodes(s.get_by_volume(p=(4.0, 5.0, 1.0)))
self.assertEqual(c, Nodes([n7]))
