mirror of
https://github.com/boostorg/ublas.git
synced 2026-02-02 21:32:09 +00:00
107 lines
3.1 KiB
C++
107 lines
3.1 KiB
C++
// Copyright (c) 2011 David Bellot
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See
|
|
// accompanying file LICENSE_1_0.txt or copy at
|
|
// http://www.boost.org/LICENSE_1_0.txt)
|
|
|
|
#include "boost/numeric/ublas/matrix_sparse.hpp"
|
|
|
|
#define BOOST_TEST_MODULE SparseMatrixErasureTest
|
|
#include <boost/test/included/unit_test.hpp>
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE( compressed_matrix_erase_after_end )
|
|
{
|
|
boost::numeric::ublas::compressed_matrix<int, boost::numeric::ublas::row_major > A(2, 2);
|
|
|
|
BOOST_CHECK_EQUAL( A.filled1(), (int) 1 );
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
|
|
A(0,0) = 1;
|
|
|
|
BOOST_CHECK_EQUAL( A.nnz(), 1 );
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
|
|
// check new element
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.index2_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.value_data()[0], 1 );
|
|
// check end of list marker
|
|
BOOST_CHECK_EQUAL( A.index1_data()[1], 1 );
|
|
|
|
A.erase_element(1,0);
|
|
|
|
BOOST_CHECK_EQUAL( A.nnz(), 1 );
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 1 );
|
|
|
|
// check new element
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.index2_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.value_data()[0], 1 );
|
|
// check end of list marker
|
|
BOOST_CHECK_EQUAL( A.index1_data()[1], 1 );
|
|
|
|
A.erase_element(0,0);
|
|
|
|
BOOST_CHECK_EQUAL( A.nnz(), 0 );
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 0 );
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE( compressed_matrix_erase_in_the_middle )
|
|
{
|
|
boost::numeric::ublas::compressed_matrix<int, boost::numeric::ublas::row_major > A(2, 2);
|
|
|
|
BOOST_CHECK_EQUAL( A.filled1(), 1 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 0 );
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
|
|
A.insert_element(0,1,5);
|
|
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 1 );
|
|
|
|
// check new element
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.index2_data()[0], 1 );
|
|
BOOST_CHECK_EQUAL( A.value_data()[0], 5 );
|
|
// check end of list marker
|
|
BOOST_CHECK_EQUAL( A.index1_data()[1], 1 );
|
|
|
|
A.insert_element(0,0,4);
|
|
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 2 );
|
|
|
|
// check new element
|
|
BOOST_CHECK_EQUAL( A.index2_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.value_data()[0], 4 );
|
|
// check previous element
|
|
BOOST_CHECK_EQUAL( A.index2_data()[1], 1 );
|
|
BOOST_CHECK_EQUAL( A.value_data()[1], 5 );
|
|
// check end of list marker
|
|
BOOST_CHECK_EQUAL( A.index1_data()[1], 2 );
|
|
|
|
A.erase_element(0,0);
|
|
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 1 );
|
|
|
|
BOOST_CHECK_EQUAL( A.index2_data()[0], 1 );
|
|
BOOST_CHECK_EQUAL( A.value_data()[0], 5 );
|
|
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.index1_data()[1], 1 );
|
|
|
|
A.erase_element(0,1);
|
|
|
|
BOOST_CHECK_EQUAL( A.filled1(), 2 );
|
|
BOOST_CHECK_EQUAL( A.filled2(), 0 );
|
|
BOOST_CHECK_EQUAL( A.index1_data()[0], 0 );
|
|
BOOST_CHECK_EQUAL( A.index1_data()[1], 0 );
|
|
|
|
}
|