testSparseVector.cc

Go to the documentation of this file.

#include <iostream>

#include <utl/SparseVector.h>

#include <tst/Verify.h>
#include <cppunit/extensions/HelperMacros.h>

using namespace std;
using namespace utl;
using namespace tst;

#define ASSERT_CLOSE(x, y) CPPUNIT_ASSERT(Verify<CloseTo>(x, y))
#define ASSERT_CLOSE_EPS(x, y, eps) CPPUNIT_ASSERT(Verify<CloseTo>(x, y, eps))
#define ASSERT_EQUAL(x, y) CPPUNIT_ASSERT(Verify<Equal>(x, y))
#define ASSERT_ITERATOR_EQUAL(it, e, i, v) \
  CPPUNIT_ASSERT(it != e); \
  ASSERT_EQUAL(it.GetIndex(), i); \
  ASSERT_EQUAL(it(), v)


class TestSparseVector : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(TestSparseVector);
  CPPUNIT_TEST(TestAccess);
  CPPUNIT_TEST(TestIterators);
  CPPUNIT_TEST(TestDotProduct);
  CPPUNIT_TEST(TestAddSubtract);
  CPPUNIT_TEST(TestScalarMultiply);
  CPPUNIT_TEST(TestCtor);
  CPPUNIT_TEST_SUITE_END();

public:
  void
  TestAccess()
  {
    // 10
    const int size = 10;
    SparseVector<double> v;

    ASSERT_EQUAL(v.GetNonEmptySize(), 0U);

    for (int i = 0; i < size; ++i)
      v[i];

    ASSERT_EQUAL(v.GetNonEmptySize(), 0U);

    for (int i = 0; i < size; ++i)
      v(i);

    ASSERT_EQUAL(v.GetNonEmptySize(), (unsigned int)(size));

    v.Reclaim();

    ASSERT_EQUAL(v.GetNonEmptySize(), 0U);

    v(5) = 1;

    ASSERT_EQUAL(v.GetNonEmptySize(), 1U);

    ASSERT_EQUAL(v[5], 1.);
  }

  void
  TestIterators()
  {
    SparseVector<int> v;
    v(0) = 1; v(10) = 2; v(100) = 3; v(1000) = -1;

    SparseVector<int>::Iterator it = v.SparseBegin();
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 0, 1);
    ++it;
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 10, 2);
    ++it;
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 100, 3);
    ++it;
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 1000, -1);
    ++it;
    CPPUNIT_ASSERT(it == v.SparseEnd());

    it = v.Find(0);
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 0, 1);

    it = v.Find(10);
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 10, 2);

    it = v.Find(1000);
    ASSERT_ITERATOR_EQUAL(it, v.SparseEnd(), 1000, -1);
  }

  void
  TestDotProduct()
  {
    SparseVector<double> a;
    a(0) = 1; a(3) = 2; a(5) = 3; a(9) = 4;

    SparseVector<double> b;
    b(0) = 1; b(1) = 0; b(2) = 2; b(3) = 3; b(4) = 4; b(6) = 6; b(7) = 7; b(9) = 9;

    const double scalar = a * b;

    ASSERT_EQUAL(scalar, 43.);
  }

  void
  TestAddSubtract()
  {
    SparseVector<double> a;
    a(0) = 1; a(3) = 2; a(5) = 3; a(9) = 4;

    SparseVector<double> b;
    b(0) = 1; b(1) = 0; b(2) = 2; b(3) = 3; b(4) = 4; b(6) = 6; b(7) = 7; b(9) = 9;

    SparseVector<double> ta(a);

    a += b;

    // a + b
    SparseVector<double> p;
    p(0) = 2; p(1) = 0; p(2) = 2; p(3) = 5; p(4) = 4; p(5) = 3; p(6) = 6; p(7) = 7; p(9) = 13;

    CPPUNIT_ASSERT(a == p);

    a.Reclaim();
    p.Reclaim();

    CPPUNIT_ASSERT(a == p);

    a -= b;

    a.Reclaim();

    CPPUNIT_ASSERT(a == ta);

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wself-assign-overloaded"
    // warning since we don't know if this is a = 0;
    a -= a;
#pragma clang diagnostic pop

    a.Reclaim();

    CPPUNIT_ASSERT(a.IsEmpty());

    ASSERT_EQUAL(a.GetNonEmptySize(), 0U);
  }

  void
  TestScalarMultiply()
  {
    SparseVector<double> a;
    a(0) = 1; a(3) = 2; a(5) = 3; a(9) = 4;

    a *= 2;

    SparseVector<double> b;
    b(0) = 2; b(3) = 4; b(5) = 6; b(9) = 8;

    CPPUNIT_ASSERT(a == b);
  }

  void
  TestCtor()
  {
    SparseVector<int> vi(1);

    SparseVector<double> vd(vi);

    CPPUNIT_ASSERT(vi == vd);
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION(TestSparseVector);