def test_grid_var_names(): model = BmiGridmet() model.initialize() names = model.get_input_var_names() assert names == ("") names = model.get_output_var_names() assert names == ( "daily_maximum_temperature", "daily_minimum_temperature", "precipitation_amount", )
import datetime def print_times(x): print(x.get_start_time()) print(x.get_time_step()) print(x.get_time_units()) print(x.get_current_time()) print(x.get_end_time()) x = BmiGridmet() # x.initialize('gridmet_bmi.yaml') x.initialize() print(x.get_input_var_names()) print(x.get_output_var_names()) grid_id = x.get_var_grid('daily_maximum_temperature') size = x.get_grid_size(grid_id) shape = np.empty(2, dtype=np.int) origin = np.empty(2, dtype=np.float) delta = np.empty(2, dtype=np.float) x.get_grid_origin(grid_id, origin) x.get_grid_spacing(grid_id, spacing=delta) tmp = x.get_grid_shape(grid_id, shape) print(type(shape), shape[0], shape[1], shape) tmp2 = np.array([585, 1386]) npt.assert_almost_equal(shape, np.array([585, 1386])) vals = np.zeros(size) x.get_value('daily_maximum_temperature', vals) print(np.nanmin(vals)) print(np.nanmax(vals))