@@ -178,5 +178,47 @@ def _unrolled_cone_curvilinear_grid():
178178 "time" : (["time" ], TIME , {"axis" : "T" }),
179179 },
180180 ),
181+ "ds_2d_right" : xr .Dataset (
182+ {
183+ "data_g" : (["time" , "ZG" , "YG" , "XG" ], np .random .rand (T , Z , Y , X )),
184+ "data_c" : (["time" , "ZC" , "YC" , "XC" ], np .random .rand (T , Z , Y , X )),
185+ "U (A grid)" : (["time" , "ZG" , "YG" , "XG" ], np .random .rand (T , Z , Y , X )),
186+ "V (A grid)" : (["time" , "ZG" , "YG" , "XG" ], np .random .rand (T , Z , Y , X )),
187+ "U (C grid)" : (["time" , "ZG" , "YC" , "XG" ], np .random .rand (T , Z , Y , X )),
188+ "V (C grid)" : (["time" , "ZG" , "YG" , "XC" ], np .random .rand (T , Z , Y , X )),
189+ },
190+ coords = {
191+ "XG" : (
192+ ["XG" ],
193+ 2 * np .pi / X * np .arange (0 , X ),
194+ {"axis" : "X" , "c_grid_axis_shift" : 0.5 },
195+ ),
196+ "XC" : (["XC" ], 2 * np .pi / X * (np .arange (0 , X ) - 0.5 ), {"axis" : "X" }),
197+ "YG" : (
198+ ["YG" ],
199+ 2 * np .pi / (Y ) * np .arange (0 , Y ),
200+ {"axis" : "Y" , "c_grid_axis_shift" : 0.5 },
201+ ),
202+ "YC" : (
203+ ["YC" ],
204+ 2 * np .pi / (Y ) * (np .arange (0 , Y ) - 0.5 ),
205+ {"axis" : "Y" },
206+ ),
207+ "ZG" : (
208+ ["ZG" ],
209+ np .arange (Z ),
210+ {"axis" : "Z" , "c_grid_axis_shift" : 0.5 },
211+ ),
212+ "ZC" : (
213+ ["ZC" ],
214+ np .arange (Z ) - 0.5 ,
215+ {"axis" : "Z" },
216+ ),
217+ "lon" : (["XG" ], 2 * np .pi / X * np .arange (0 , X )),
218+ "lat" : (["YG" ], 2 * np .pi / (Y ) * np .arange (0 , Y )),
219+ "depth" : (["ZG" ], np .arange (Z )),
220+ "time" : (["time" ], TIME , {"axis" : "T" }),
221+ },
222+ ),
181223 "2d_left_unrolled_cone" : _unrolled_cone_curvilinear_grid (),
182224}
0 commit comments