Facetgrid

cf_xarray/accessor.py

@@ -30,6 +30,7 @@
     xr.core.rolling.Rolling,
     xr.core.rolling.Coarsen,
     xr.core.weighted.Weighted,
+    xr.plot.facetgrid.FacetGrid,
 )
 
 #:  `axis` names understood by cf_xarray
@@ -459,7 +460,7 @@ def _getattr(
     attr: str,
     accessor: "CFAccessor",
     key_mappers: Mapping[str, Mapper],
-    wrap_classes: bool = False,
+    wrap_classes: bool = True,
     extra_decorator: Callable = None,
 ):
     """

cf_xarray/tests/test_accessor.py

@@ -294,24 +294,10 @@ def test_dataarray_plot(obj):
     assert isinstance(rv, mpl.contour.QuadContourSet)
     plt.close()
 
-    rv = obj.cf.plot(x="X", y="Y", col="T")
-    assert isinstance(rv, xr.plot.FacetGrid)
-    plt.close()
-
-    rv = obj.cf.plot.contourf(x="X", y="Y", col="T")
-    assert isinstance(rv, xr.plot.FacetGrid)
-    plt.close()
-
     rv = obj.isel(lat=[0, 1], lon=1).cf.plot.line(x="T", hue="Y")
     assert all([isinstance(line, mpl.lines.Line2D) for line in rv])
     plt.close()
 
-    obj = obj.copy(deep=True)
-    obj.time.attrs.clear()
-    rv = obj.cf.plot(x="X", y="Y", col="time")
-    assert isinstance(rv, xr.plot.FacetGrid)
-    plt.close()
-
 
 @pytest.mark.parametrize("obj", datasets)
 def test_dataset_plot(obj):