Ejemplo n.º 1
0
    def test_raster_field_3d (self):
        g = RasterField ((4, 3, 2), (1.5, 1., 3), (0.5, 0, -.5 ),
                         indexing='ij')
        self.assert_point_count (g, 24)
        self.assert_cell_count (g, 6)

        g.add_field ('Point Data', g.get_x (), centering='point')

        cell_data = np.arange (6.)*10.
        g.add_field ('Cell Data', cell_data, centering='zonal')

        g.get_field ('Cell Data')
        self.assert_field_values (g, 'Cell Data',
                                  [0., 10., 20., 30., 40., 50.])

        self.assert_field_values (g, 'Point Data',
                                  [-0.5, 2.5, -0.5, 2.5, -0.5, 2.5,
                                   -0.5, 2.5, -0.5, 2.5, -0.5, 2.5,
                                   -0.5, 2.5, -0.5, 2.5, -0.5, 2.5,
                                   -0.5, 2.5, -0.5, 2.5, -0.5, 2.5])

        self.assert_shape (g, (4, 3, 2))

        x = g.get_x ()
        data = g.get_field ('Point Data')

        self.assertEqual (x.size, data.size)
Ejemplo n.º 2
0
    def test_raster_field_3d(self):
        g = RasterField((4, 3, 2), (1.5, 1.0, 3), (0.5, 0, -0.5),
                        indexing="ij")
        self.assert_point_count(g, 24)
        self.assert_cell_count(g, 6)

        g.add_field("Point Data", g.get_x(), centering="point")

        cell_data = np.arange(6.0) * 10.0
        g.add_field("Cell Data", cell_data, centering="zonal")

        g.get_field("Cell Data")
        self.assert_field_values(g, "Cell Data",
                                 [0.0, 10.0, 20.0, 30.0, 40.0, 50.0])

        self.assert_field_values(
            g,
            "Point Data",
            [
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
            ],
        )

        self.assert_shape(g, (4, 3, 2))

        x = g.get_x()
        data = g.get_field("Point Data")

        self.assertEqual(x.size, data.size)
Ejemplo n.º 3
0
    def test_raster_field_3d(self):
        g = RasterField((4, 3, 2), (1.5, 1.0, 3), (0.5, 0, -0.5), indexing="ij")
        self.assert_point_count(g, 24)
        self.assert_cell_count(g, 6)

        g.add_field("Point Data", g.get_x(), centering="point")

        cell_data = np.arange(6.0) * 10.0
        g.add_field("Cell Data", cell_data, centering="zonal")

        g.get_field("Cell Data")
        self.assert_field_values(g, "Cell Data", [0.0, 10.0, 20.0, 30.0, 40.0, 50.0])

        self.assert_field_values(
            g,
            "Point Data",
            [
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
                -0.5,
                2.5,
            ],
        )

        self.assert_shape(g, (4, 3, 2))

        x = g.get_x()
        data = g.get_field("Point Data")

        self.assertEqual(x.size, data.size)