def test_empty(self): resource = os.path.join(os.path.dirname(__file__), 'resources', 'empty.txt') output_path = os.path.join(os.path.dirname(__file__), 'resources', 'empty.nc') w = WeraAsciiTotals(resource) assert w.data.empty with self.assertRaises(ValueError): w.export(output_path)
def test_export_skio_netcdf(self): output_path = os.path.join(os.path.dirname(__file__), 'resources', 'skio_ij.nc') w = WeraAsciiTotals(self.resource) w.export(output_path) with netCDF4.Dataset(output_path) as nc: assert np.isclose(nc.variables['u'][0, 14, 52], 0.009) assert np.isclose(nc.variables['v'][0, 14, 52], -0.307) assert np.isclose(nc.variables['uacc'][0, 14, 52], 0.050) assert np.isclose(nc.variables['vacc'][0, 14, 52], 0.027) assert np.isclose(nc.variables['u'][0, 58, 56], 0.363) assert np.isclose(nc.variables['v'][0, 58, 56], 0.629) assert np.isclose(nc.variables['uacc'][0, 58, 56], 0.072) assert np.isclose(nc.variables['vacc'][0, 58, 56], 0.270)
def test_export_netcdf(self): resource = os.path.join(os.path.dirname(__file__), 'resources', 'totals.txt') output_path = os.path.join(os.path.dirname(__file__), 'resources', 'totals.nc') w = WeraAsciiTotals(resource) w.export(output_path) with netCDF4.Dataset(output_path) as nc: assert np.isclose(nc.variables['u'][0, 91, 100], -0.043) assert np.isclose(nc.variables['v'][0, 91, 100], -0.047) assert np.isclose(nc.variables['uacc'][0, 91, 100], 0.036) assert np.isclose(nc.variables['vacc'][0, 91, 100], 0.037) assert np.isclose(nc.variables['u'][0, 101, 88], -0.067) assert np.isclose(nc.variables['v'][0, 101, 88], -0.058) assert np.isclose(nc.variables['uacc'][0, 101, 88], 0.035) assert np.isclose(nc.variables['vacc'][0, 101, 88], 0.025)
def test_one_reporting(self): resource = os.path.join(os.path.dirname(__file__), 'resources', 'onereporting.txt') output_path = os.path.join(os.path.dirname(__file__), 'resources', 'onereporting.nc') w = WeraAsciiTotals(resource) assert not w.data.empty w.export(output_path) with netCDF4.Dataset(output_path) as nc: assert np.isclose(nc.variables['u'][0, 68, 70], 0) assert np.isclose(nc.variables['v'][0, 68, 70], 0.) assert np.isclose(nc.variables['uacc'][0, 68, 70], 0.035) assert np.isclose(nc.variables['vacc'][0, 68, 70], 0.117) assert np.isclose(nc.variables['u'][0, 69, 107], 0.014) assert np.isclose(nc.variables['v'][0, 69, 107], -0.103) assert np.isclose(nc.variables['uacc'][0, 69, 107], 0.051) assert np.isclose(nc.variables['vacc'][0, 69, 107], 0.095)