Coverage for model_plots/bar.py: 64%

1import xarray as xr 

2from dantro.plot.funcs.generic import make_facet_grid_plot 

3 

4from utopya.eval import PlotHelper 

5 

6 

7@make_facet_grid_plot( 

8 map_as="dataset", 

9 encodings=("col", "row"), 

10 supported_hue_styles=("discrete",), 

11 hue_style="discrete", 

12 add_guide=False, 

13 register_as_kind=True, 

14 overwrite_existing=True, 

15) 

16def bar( 

17 ds: xr.Dataset, 

18 hlpr: PlotHelper, 

19 *, 

20 x: str, 

21 y: str, 

22 hue: str = None, 

23 _is_facetgrid: bool, 

24 **plot_kwargs, 

25): 

26 if "width" not in plot_kwargs: 

27 plot_kwargs["width"] = ds[x][1] - ds[x][0] 

28 

29 hlpr.ax.bar(ds[x], ds[y], **plot_kwargs) 

30 

31 

32@make_facet_grid_plot( 

33 map_as="dataarray", 

34 encodings=("col", "row"), 

35 supported_hue_styles=("discrete",), 

36 hue_style="discrete", 

37 add_guide=False, 

38 register_as_kind=True, 

39 overwrite_existing=True, 

40 drop_kwargs=("x", "y"), 

41) 

42def hist( 

43 ds: xr.Dataset, 

44 hlpr: PlotHelper, 

45 *, 

46 _is_facetgrid: bool, 

47 **plot_kwargs, 

48): 

49 hlpr.ax.hist(ds, **plot_kwargs)