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)
