Include bbox() functions for YTCoveringGrids and YTIntersectionContainer3D

yt/data_objects/construction_data_containers.py

@@ -875,6 +875,9 @@ def _setup_data_source(self):
         # their cell centers.
         self._data_source.loose_selection = True
 
+    def get_bbox(self):
+        return (self.left_edge, self.right_edge)
+
     def get_data(self, fields=None):
         if fields is None:
             return

yt/data_objects/selection_objects/boolean_operations.py

@@ -102,7 +102,17 @@ def __init__(self, data_objects, ds=None, field_parameters=None, data_source=Non
         validate_object(data_source, YTSelectionContainer)
         YTSelectionContainer3D.__init__(self, None, ds, field_parameters, data_source)
         self.data_objects = list(always_iterable(data_objects))
-
+
+    def get_bbox(self):
+        # Get the bounding box of the intersection
+        bbox = self.data_objects[0].get_bbox()
+        for obj in self.data_objects[1:]:
+            cur_bbox = obj._get_bbox()
+            bbox = (
+                np.maximum(bbox[0], cur_bbox[0]),
+                np.minimum(bbox[1], cur_bbox[1]),
+        )
+        return bbox
 
 class YTDataObjectUnion(YTSelectionContainer3D):
     """

yt/data_objects/tests/test_bbox.py

@@ -3,7 +3,7 @@
 import numpy as np
 from numpy.testing import assert_equal
 
-from yt.testing import assert_allclose_units, fake_amr_ds
+from yt.testing import assert_allclose_units, fake_amr_ds, fake_octree_ds
 
 
 def test_object_bbox():
@@ -55,3 +55,29 @@ def test_object_bbox():
     assert_allclose_units(re3, re0)
     assert_equal(le4, regb.left_edge)
     assert_equal(re4, regb.right_edge)
+
+def test_covering_grid_bbox():
+    ds = fake_octree_ds(num_zones=32)
+
+    cg = ds.covering_grid(level=0, left_edge=[0.3, 0.3, 0.3], dims=[8, 8, 8])
+
+    # Make a covering grid with a data source
+    sp = ds.sphere(
+        [0.5, 0.5, 0.5],
+        0.15,
+    )
+    cgds = ds.covering_grid(
+        level=0, left_edge=[0.3, 0.3, 0.3], dims=[8, 8, 8], data_source = sp,
+    )
+
+    # The bounding boxes of the two covering grids should be the same
+    cg_bbox = cg.get_bbox()
+    cgds_bbox = cgds.get_bbox()
+    assert_equal(cg_bbox, cgds_bbox)
+
+    # The bounding box of the cg's data source should be the left edge of sp and right edge of cgds
+    cgds_ds_bbox = cgds._data_source.get_bbox()
+    le_sp, re_sp = sp.get_bbox()
+
+    assert_equal(le_sp, cgds_ds_bbox[0])
+    assert_equal(cgds_ds_bbox[1], cgds_ds_bbox[1])