def test_nodes_at_diagonal_1d(): """Test nodes at diagonals for 1d grid.""" diagonals = create_nodes_at_diagonal((1, 2)) assert_array_equal(diagonals, np.array([], dtype=int).reshape((0, 2))) diagonals = create_nodes_at_diagonal((4, 1)) assert_array_equal(diagonals, np.array([], dtype=int).reshape((0, 2)))
def test_nodes_at_diagonal_1d(): """Test nodes at diagonals for 1d grid.""" diagonals = create_nodes_at_diagonal((1, 2)) assert_array_equal(diagonals, np.array([], dtype=int).reshape((0, 2))) diagonals = create_nodes_at_diagonal((4, 1)) assert_array_equal(diagonals, np.array([], dtype=int).reshape((0, 2)))
def test_nodes_at_diagonal(): """Test tail and head nodes of diagonals.""" diagonals = create_nodes_at_diagonal((2, 2)) assert_array_equal(diagonals, [[0, 3], [1, 2]]) diagonals = create_nodes_at_diagonal((4, 3)) assert_is_instance(diagonals, np.ndarray) assert_array_equal(diagonals, [[0, 4], [1, 3], [1, 5], [2, 4], [3, 7], [4, 6], [4, 8], [5, 7], [6, 10], [7, 9], [7, 11], [8, 10]])
def test_nodes_at_diagonal(): """Test tail and head nodes of diagonals.""" diagonals = create_nodes_at_diagonal((2, 2)) assert_array_equal(diagonals, [[0, 3], [1, 2]]) diagonals = create_nodes_at_diagonal((4, 3)) assert_is_instance(diagonals, np.ndarray) assert_array_equal(diagonals, [[0, 4], [1, 3], [1, 5], [2, 4], [3, 7], [4, 6], [4, 8], [5, 7], [6, 10], [7, 9], [7, 11], [8, 10]])
def test_nodes_at_diagonal_out_keyword(): """Test out keyword for nodes_at_diagonal.""" buffer = np.empty((4, 2), dtype=int) diagonals = create_nodes_at_diagonal((3, 2), out=buffer) assert buffer is diagonals assert_array_equal(diagonals, [[0, 3], [1, 2], [2, 5], [3, 4]])
def test_nodes_at_diagonal_out_keyword(): """Test out keyword for nodes_at_diagonal.""" buffer = np.empty((4, 2), dtype=int) diagonals = create_nodes_at_diagonal((3, 2), out=buffer) assert_is(buffer, diagonals) assert_array_equal(diagonals, [[0, 3], [1, 2], [2, 5], [3, 4]])