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]]))
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]]))
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']
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"])
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"])
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']
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'])
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'])
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], ] ), )
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)))