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_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))