Ejemplo n.º 1
0
 def test_boundary_node_mask(self):
     diags = sgrid.diagonal_node_array((3, 3), out_of_bounds=-1,
                                       boundary_node_mask=-2)
     self.assertArrayEqual(diags, 
                        np.array([[-2, -2, -2, -2], [-2, -2, -2, -2], [-2, -2, -2, -2],
                                  [-2, -2, -2, -2], [ 8,  6,  0,  2], [-2, -2, -2, -2],
                                  [-2, -2, -2, -2], [-2, -2, -2, -2], [-2, -2, -2, -2]]))
Ejemplo n.º 2
0
def test_boundary_node_mask():
    diags = sgrid.diagonal_node_array((3, 3),
                                      out_of_bounds=-1,
                                      boundary_node_mask=-2)
    assert_array_equal(
        diags,
        np.array([[-2, -2, -2, -2], [-2, -2, -2, -2], [-2, -2, -2, -2],
                  [-2, -2, -2, -2], [8, 6, 0, 2], [-2, -2, -2, -2],
                  [-2, -2, -2, -2], [-2, -2, -2, -2], [-2, -2, -2, -2]]))
Ejemplo n.º 3
0
def test_default():
    diags = sgrid.diagonal_node_array((2, 3), out_of_bounds=-1)
    assert_array_equal(
        diags,
        np.array([[4, -1, -1, -1], [5, 3, -1, -1], [-1, 4, -1, -1],
                  [-1, -1, -1, 1], [-1, -1, 0, 2], [-1, -1, 1, -1]]))

    assert diags.base is None
    assert diags.flags['C_CONTIGUOUS']
Ejemplo n.º 4
0
def test_non_contiguous():
    diags = sgrid.diagonal_node_array((2, 3), out_of_bounds=-1, contiguous=False)
    assert_array_equal(
        diags,
        np.array([[4, -1, -1, -1], [5, 3, -1, -1], [-1, 4, -1, -1], [-1, -1, -1, 1], [-1, -1, 0, 2], [-1, -1, 1, -1]]),
    )

    assert_true(isinstance(diags.base, np.ndarray))
    assert_false(diags.flags["C_CONTIGUOUS"])
Ejemplo n.º 5
0
def test_default():
    diags = sgrid.diagonal_node_array((2, 3), out_of_bounds=-1)
    assert_array_equal(
        diags,
        np.array([[4, -1, -1, -1], [5, 3, -1, -1], [-1, 4, -1, -1], [-1, -1, -1, 1], [-1, -1, 0, 2], [-1, -1, 1, -1]]),
    )

    assert_true(diags.base is None)
    assert_true(diags.flags["C_CONTIGUOUS"])
Ejemplo n.º 6
0
def test_non_contiguous():
    diags = sgrid.diagonal_node_array((2, 3),
                                      out_of_bounds=-1,
                                      contiguous=False)
    assert_array_equal(
        diags,
        np.array([[4, -1, -1, -1], [5, 3, -1, -1], [-1, 4, -1, -1],
                  [-1, -1, -1, 1], [-1, -1, 0, 2], [-1, -1, 1, -1]]))

    assert isinstance(diags.base, np.ndarray)
    assert not diags.flags['C_CONTIGUOUS']
Ejemplo n.º 7
0
    def test_default(self):
        diags = sgrid.diagonal_node_array((2, 3), out_of_bounds=-1)
        self.assertArrayEqual(diags,
                           np.array([[ 4, -1, -1, -1],
                                     [ 5,  3, -1, -1],
                                     [-1,  4, -1, -1],
                                     [-1, -1, -1,  1],
                                     [-1, -1,  0,  2],
                                     [-1, -1,  1, -1]]))

        self.assertTrue(diags.base is None)
        self.assertTrue(diags.flags['C_CONTIGUOUS'])
Ejemplo n.º 8
0
    def test_non_contiguous(self):
        diags = sgrid.diagonal_node_array((2, 3), out_of_bounds=-1,
                                          contiguous=False)
        self.assertArrayEqual(diags,
                           np.array([[ 4, -1, -1, -1],
                                     [ 5,  3, -1, -1],
                                     [-1,  4, -1, -1],
                                     [-1, -1, -1,  1],
                                     [-1, -1,  0,  2],
                                     [-1, -1,  1, -1]]))

        self.assertTrue(isinstance(diags.base, np.ndarray))
        self.assertFalse(diags.flags['C_CONTIGUOUS'])
Ejemplo n.º 9
0
def test_boundary_node_mask():
    diags = sgrid.diagonal_node_array((3, 3), out_of_bounds=-1, boundary_node_mask=-2)
    assert_array_equal(
        diags,
        np.array(
            [
                [-2, -2, -2, -2],
                [-2, -2, -2, -2],
                [-2, -2, -2, -2],
                [-2, -2, -2, -2],
                [8, 6, 0, 2],
                [-2, -2, -2, -2],
                [-2, -2, -2, -2],
                [-2, -2, -2, -2],
                [-2, -2, -2, -2],
            ]
        ),
    )
Ejemplo n.º 10
0
def test_boundary_node_mask_no_actives():
    diags = sgrid.diagonal_node_array((2, 3),
                                      out_of_bounds=-1,
                                      boundary_node_mask=-2)
    assert_array_equal(diags, -2 * np.ones((6, 4)))
Ejemplo n.º 11
0
def test_boundary_node_mask_no_actives():
    diags = sgrid.diagonal_node_array((2, 3), out_of_bounds=-1, boundary_node_mask=-2)
    assert_array_equal(diags, -2 * np.ones((6, 4)))