def test_diagnostic_vars(self): d = WRF(get_demo_file('wrf_tip_d1.nc')) d2 = GeoNetcdf(get_demo_file('wrf_tip_d2.nc')) self.assertTrue('T2C' in d.variables) ref = d.get_vardata('T2') tot = d.get_vardata('T2C') + 273.15 np.testing.assert_allclose(ref, tot) d.set_roi(grid=d2.grid) d.set_subset(toroi=True) ref = d.get_vardata('T2') tot = d.get_vardata('T2C') + 273.15 self.assertEqual(tot.shape[-1] * 3, d2.grid.nx) self.assertEqual(tot.shape[-2] * 3, d2.grid.ny) np.testing.assert_allclose(ref, tot) d = WRF(get_demo_file('wrf_tip_d1.nc')) ref = d.variables['T2'][:] d.set_subset(margin=-5) tot = d.get_vardata('T2') assert_array_equal(ref.shape[1] - 10, tot.shape[1]) assert_array_equal(ref.shape[2] - 10, tot.shape[2]) assert_array_equal(ref[:, 5:-5, 5:-5], tot)
def test_diagnostic_vars(self): d = WRF(get_demo_file("wrf_tip_d1.nc")) d2 = GeoNetcdf(get_demo_file("wrf_tip_d2.nc")) self.assertTrue("T2C" in d.variables) ref = d.get_vardata("T2") tot = d.get_vardata("T2C") + 273.15 np.testing.assert_allclose(ref, tot) d.set_roi(grid=d2.grid) d.set_subset(toroi=True) ref = d.get_vardata("T2") tot = d.get_vardata("T2C") + 273.15 self.assertEqual(tot.shape[-1] * 3, d2.grid.nx) self.assertEqual(tot.shape[-2] * 3, d2.grid.ny) np.testing.assert_allclose(ref, tot) d = WRF(get_demo_file("wrf_tip_d1.nc")) ref = d.variables["T2"][:] d.set_subset(margin=-5) tot = d.get_vardata("T2") assert_array_equal(ref.shape[1] - 10, tot.shape[1]) assert_array_equal(ref.shape[2] - 10, tot.shape[2]) assert_array_equal(ref[:, 5:-5, 5:-5], tot)