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]))