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)