def test_is_rectangular(self):
sub_box_1 = selection.SelectionBox((0, 0, 0), (5, 5, 5))
box_1 = selection.SelectionGroup((sub_box_1, ))
self.assertTrue(box_1.is_rectangular)
sub_box_2 = selection.SelectionBox((0, 5, 0), (5, 10, 5))
box_1.add_box(sub_box_2)
self.assertTrue(box_1.is_rectangular)
def test_is_contiguous(self):
sub_box_1 = selection.SelectionBox((0, 0, 0), (5, 5, 5))
box_1 = selection.SelectionGroup((sub_box_1, ))
self.assertTrue(box_1.is_contiguous)
sub_box_2 = selection.SelectionBox((5, 5, 5), (10, 10, 10))
box_1.add_box(sub_box_2)
self.assertTrue(box_1.is_contiguous)
sub_box_3 = selection.SelectionBox((6, 6, 6), (10, 10, 10))
box_2 = selection.SelectionGroup((sub_box_1, sub_box_3))
self.assertFalse(box_2.is_contiguous)
def test_intersects(self):
box_1 = selection.SelectionBox((0, 0, 0), (5, 5, 5))
box_2 = selection.SelectionBox((4, 4, 4), (10, 10, 10))
self.assertTrue(box_1.intersects(box_2))
self.assertTrue(box_2.intersects(box_1))
box_3 = selection.SelectionBox((5, 5, 5), (10, 10, 10))
self.assertFalse(box_1.intersects(box_3))
self.assertFalse(box_3.intersects(box_1))
box_4 = selection.SelectionBox((1, 20, 1), (4, 25, 4))
self.assertFalse(box_1.intersects(box_4))
self.assertFalse(box_4.intersects(box_1))
def test_sorted_iterator(self):
sub_box_1 = selection.SelectionBox((0, 0, 0), (4, 4, 4))
box_1 = selection.SelectionGroup((sub_box_1, ))
box_2 = selection.SelectionGroup((sub_box_1, ))
sub_box_2 = selection.SelectionBox((7, 7, 7), (10, 10, 10))
box_1.add_box(sub_box_2)
sub_box_3 = selection.SelectionBox((15, 15, 15), (20, 20, 20))
box_2.add_box(sub_box_3)
box_1.add_box(sub_box_3)
box_2.add_box(sub_box_2)
for sb1, sb2 in zip(box_1.selection_boxes, box_2.selection_boxes):
self.assertEqual(sb1.shape, sb2.shape)
def test_single_block_box(self):
sub_box_1 = selection.SelectionBox((0, 0, 0), (1, 1, 2))
box_1 = selection.SelectionGroup((sub_box_1, ))
self.assertEqual((1, 1, 2), sub_box_1.shape)
self.assertEqual(2, len([x for x in sub_box_1]))
self.assertIn((0, 0, 0), sub_box_1)
self.assertIn((1, 1, 2), sub_box_1)
self.assertNotIn((1, 1, 3), sub_box_1)
def test_add_box(self): # Quick crude test, needs more cases
sub_box_1 = selection.SelectionBox((0, 0, 0), (5, 5, 5))
box_1 = selection.SelectionGroup((sub_box_1, ))
self.assertEqual(1, len(box_1))
box_1.add_box(selection.SelectionBox((0, 5, 0), (5, 10, 5)))
self.assertEqual(1, len(box_1))
box_1.add_box(selection.SelectionBox((0, 10, 0), (5, 15, 5)))
self.assertEqual(1, len(box_1))
box_2 = selection.SelectionGroup((sub_box_1, ))
self.assertEqual(1, len(box_2))
box_2.add_box(selection.SelectionBox((0, 6, 0), (5, 10, 5)))
self.assertEqual(2, len(box_2))
box_3 = selection.SelectionGroup((sub_box_1, ))
self.assertEqual(1, len(box_3))
box_3.add_box(selection.SelectionBox((0, 10, 0), (5, 15, 5)))
self.assertEqual(2, len(box_3))
box_3.add_box(selection.SelectionBox((0, 5, 0), (5, 10, 5)))
self.assertEqual(1, len(box_3))