def test_tri_example(tmpdir):
g = GridField([0, 2, 1, 3], [0, 0, 1, 1], [0, 2, 1, 2, 3, 1], [3, 6])
g.add_field("Elevation", [1.0, 2.0, 3, 4], centering="point")
g.add_field("Temperature", [10.0, 20.0, 30.0, 40.0], centering="point")
g.add_field("Cell Elevation", [1.0, 2.0], centering="zonal")
g.add_field("Cell Temperature", [10.0, 20.0], centering="zonal")
with tmpdir.as_cwd():
field_tofile(g, "tri.vtu", format="appended", encoding="base64")
assert os.path.isfile("tri.vtu")
def test_0d(self):
vtk_file = self.vtk_files["test_0d"]
field = RasterField((1, ), (0, ), (0, ), indexing="ij", units=("m", ))
attrs = dict(description="Example 0D nc file", author="Eric")
for i in range(10):
field.add_field("Elevation", i * 10.0, centering="point")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_1d(self):
vtk_file = self.vtk_files['test_1d']
field = RasterField((12, ), (1, ), (-1, ), indexing='ij', units=('m', ))
data = np.arange(12.)
field.add_field('Elevation', data, centering='point')
attrs = dict(description='Example 1D nc file', author='Eric')
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_tri_example(tmpdir):
g = GridField([0, 2, 1, 3], [0, 0, 1, 1], [0, 2, 1, 2, 3, 1], [3, 6])
g.add_field("Elevation", [1.0, 2.0, 3, 4], centering="point")
g.add_field("Temperature", [10.0, 20.0, 30.0, 40.0], centering="point")
g.add_field("Cell Elevation", [1.0, 2.0], centering="zonal")
g.add_field("Cell Temperature", [10.0, 20.0], centering="zonal")
with tmpdir.as_cwd():
field_tofile(g, "tri.vtu", format="appended", encoding="base64")
assert os.path.isfile("tri.vtu")
def test_0d(self):
vtk_file = self.vtk_files["test_0d"]
field = RasterField((1,), (0,), (0,), indexing="ij", units=("m",))
attrs = dict(description="Example 0D nc file", author="Eric")
for i in range(10):
field.add_field("Elevation", i * 10.0, centering="point")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_1d(self):
vtk_file = self.vtk_files["test_1d"]
field = RasterField((12,), (1,), (-1,), indexing="ij", units=("m",))
data = np.arange(12.0)
field.add_field("Elevation", data, centering="point")
attrs = dict(description="Example 1D nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_1d(self):
vtk_file = self.vtk_files["test_1d"]
field = UnstructuredField(
[0, 1, 4, 10], [0, 1, 1, 2, 2, 3], [2, 4, 6], units=("m",)
)
field.add_field("point_field", np.arange(4), centering="point", units="C")
field.add_field("cell_field", np.arange(3) * 10.0, centering="zonal", units="F")
field_tofile(field, vtk_file, append=False)
def test_1d(self):
vtk_file = self.vtk_files["test_1d"]
field = RectilinearField((1, 2, 4, 5), indexing="ij", units=("m", ))
data = np.arange(4.0)
field.add_field("Elevation", data, centering="point")
attrs = dict(description="Example 1D nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_0d(self):
vtk_file = self.vtk_files['test_0d']
field = RasterField((1, ), (0, ), (0, ), indexing='ij',
units=('m', ))
attrs = dict(description='Example 0D nc file', author='Eric')
for i in range(10):
field.add_field('Elevation', i * 10., centering='point')
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_1d(self):
vtk_file = self.vtk_files['test_1d']
field = UnstructuredField([0, 1, 4, 10], [0, 1, 1, 2, 2, 3],
[2, 4, 6], units=('m', ))
field.add_field('point_field', np.arange(4), centering='point',
units='C')
field.add_field('cell_field', np.arange(3) * 10., centering='zonal',
units='F')
field_tofile(field, vtk_file, append=False)
def test_2d(self):
vtk_file = self.vtk_files['test_2d']
field = UnstructuredField([0, 2, 1, 3], [0, 0, 1, 1],
[0, 2, 1, 2, 3, 1], [3, 6],
units=('m', 'm'))
data = np.arange(4)
field.add_field('point_field', data, centering='point', units='m')
data = np.arange(2)
field.add_field('cell_field', data, centering='zonal', units='m^2')
field_tofile(field, vtk_file, append=False)
def test_2d(self):
vtk_file = self.vtk_files["test_2d"]
field = UnstructuredField([0, 2, 1, 3], [0, 0, 1, 1],
[0, 2, 1, 2, 3, 1], [3, 6],
units=("m", "m"))
data = np.arange(4)
field.add_field("point_field", data, centering="point", units="m")
data = np.arange(2)
field.add_field("cell_field", data, centering="zonal", units="m^2")
field_tofile(field, vtk_file, append=False)
def test_2d(self):
vtk_file = self.vtk_files["test_2d"]
field = UnstructuredField(
[0, 2, 1, 3], [0, 0, 1, 1], [0, 2, 1, 2, 3, 1], [3, 6], units=("m", "m")
)
data = np.arange(4)
field.add_field("point_field", data, centering="point", units="m")
data = np.arange(2)
field.add_field("cell_field", data, centering="zonal", units="m^2")
field_tofile(field, vtk_file, append=False)
def test_3d(self):
vtk_file = self.vtk_files['test_3d']
field = RasterField((2, 3, 4), (1, 2, 3), (-1, 0, 1),
indexing='ij', units=('mm', 'm', 'km'))
data = np.arange(24.)
field.add_field('Temperature', data * 10, centering='point', units='C')
field.add_field('Elevation', data, centering='point', units='meters')
field.add_field('Velocity', data * 100, centering='point', units='m/s')
field.add_field('Temp', data * 2, centering='point', units='F')
attrs = dict(description='Example 3D nc file', author='Eric')
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_1d(self):
vtk_file = self.vtk_files["test_1d"]
field = UnstructuredField([0, 1, 4, 10], [0, 1, 1, 2, 2, 3], [2, 4, 6],
units=("m", ))
field.add_field("point_field",
np.arange(4),
centering="point",
units="C")
field.add_field("cell_field",
np.arange(3) * 10.0,
centering="zonal",
units="F")
field_tofile(field, vtk_file, append=False)
def test_3d(self):
vtk_file = self.vtk_files["test_3d"]
field = RasterField(
(2, 3, 4), (1, 2, 3), (-1, 0, 1), indexing="ij", units=("mm", "m", "km")
)
data = np.arange(24.0)
field.add_field("Temperature", data * 10, centering="point", units="C")
field.add_field("Elevation", data, centering="point", units="meters")
field.add_field("Velocity", data * 100, centering="point", units="m/s")
field.add_field("Temp", data * 2, centering="point", units="F")
attrs = dict(description="Example 3D nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_3d(self):
vtk_file = self.vtk_files["test_3d"]
field = RasterField((2, 3, 4), (1, 2, 3), (-1, 0, 1),
indexing="ij",
units=("mm", "m", "km"))
data = np.arange(24.0)
field.add_field("Temperature", data * 10, centering="point", units="C")
field.add_field("Elevation", data, centering="point", units="meters")
field.add_field("Velocity", data * 100, centering="point", units="m/s")
field.add_field("Temp", data * 2, centering="point", units="F")
attrs = dict(description="Example 3D nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_2d(self):
vtk_file = self.vtk_files['test_2d']
field = RectilinearField(
(1, 4, 5), (2, 3), indexing='ij',
units=('degrees_north', 'degrees_east'),
coordinate_names=['latitude', 'longitude'])
data = np.arange(6.)
field.add_field('Temperature', data*10, centering='point',
units='C')
field.add_field('Elevation', data, centering='point',
units='meters')
field.add_field('Velocity', data*100, centering='point',
units='m/s')
field.add_field('Temp', data*2, centering='point', units='F')
attrs = dict(description='Example nc file', author='Eric')
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_2d(self):
vtk_file = self.vtk_files["test_2d"]
field = RectilinearField(
(1, 4, 5),
(2, 3),
indexing="ij",
units=("degrees_north", "degrees_east"),
coordinate_names=["latitude", "longitude"],
)
data = np.arange(6.0)
field.add_field("Temperature", data * 10, centering="point", units="C")
field.add_field("Elevation", data, centering="point", units="meters")
field.add_field("Velocity", data * 100, centering="point", units="m/s")
field.add_field("Temp", data * 2, centering="point", units="F")
attrs = dict(description="Example nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_2d(self):
vtk_file = self.vtk_files["test_2d"]
field = StructuredField(
(1, 4, 5, 1, 4, 5),
(2, 3, 2, 3, 2, 3),
(3, 2),
indexing="ij",
units=("m", "km"),
)
data = np.arange(6.0)
field.add_field("Temperature", data * 10, centering="point", units="C")
field.add_field("Elevation", data, centering="point", units="meters")
field.add_field("Velocity", data * 100, centering="point", units="m/s")
field.add_field("Temp", data * 2, centering="point", units="F")
attrs = dict(description="Example nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_2d(self):
vtk_file = self.vtk_files["test_2d"]
field = RectilinearField(
(1, 4, 5),
(2, 3),
indexing="ij",
units=("degrees_north", "degrees_east"),
coordinate_names=["latitude", "longitude"],
)
data = np.arange(6.0)
field.add_field("Temperature", data * 10, centering="point", units="C")
field.add_field("Elevation", data, centering="point", units="meters")
field.add_field("Velocity", data * 100, centering="point", units="m/s")
field.add_field("Temp", data * 2, centering="point", units="F")
attrs = dict(description="Example nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))
def test_2d(self):
vtk_file = self.vtk_files["test_2d"]
field = StructuredField(
(1, 4, 5, 1, 4, 5),
(2, 3, 2, 3, 2, 3),
(3, 2),
indexing="ij",
units=("m", "km"),
)
data = np.arange(6.0)
field.add_field("Temperature", data * 10, centering="point", units="C")
field.add_field("Elevation", data, centering="point", units="meters")
field.add_field("Velocity", data * 100, centering="point", units="m/s")
field.add_field("Temp", data * 2, centering="point", units="F")
attrs = dict(description="Example nc file", author="Eric")
field_tofile(field, vtk_file, attrs=attrs, append=True)
self.assertTrue(os.path.isfile(vtk_file))