package com.googlecode.gaal.data.impl;
import static org.junit.Assert.*;
import org.junit.Before;
import org.junit.Test;
import com.googlecode.gaal.data.api.Vector;
public class SparseVectorTest {
@Before
public void setUp() throws Exception {
}
@Test
public void testSet() {
Vector vector = new SparseVector();
vector.set(10, 1);
vector.set(20, 2);
vector.set(10, 3);
vector.set(20, 4);
System.out.println(vector);
assertTrue(vector.get(10) == 3);
assertTrue(vector.get(20) == 4);
}
@Test
public void testAdd() {
Vector vector = new SparseVector();
vector.add(10, 1);
vector.add(20, 2);
vector.add(10, 3);
vector.add(20, 4);
System.out.println(vector);
assertTrue(vector.get(10) == 4);
assertTrue(vector.get(20) == 6);
}
@Test
public void testGet() {
Vector vector = new SparseVector();
vector.add(10, 1);
vector.add(20, 2);
System.out.println(vector);
assertTrue(vector.get(10) == 1);
assertTrue(vector.get(20) == 2);
}
@Test
public void testNorm() {
Vector vector = new SparseVector();
vector.add(10, 3);
vector.add(20, 4);
System.out.println(vector);
assertTrue(vector.norm() == 5);
vector.add(10, 3);
vector.add(20, 4);
System.out.println(vector);
assertTrue(vector.norm() == 10);
}
@Test
public void testSimilarity() {
Vector vector1 = new SparseVector();
vector1.add(10, 3);
vector1.add(20, 4);
System.out.println(vector1);
Vector vector2 = new SparseVector();
vector2.add(10, 3);
vector2.add(20, 4);
System.out.println(vector2);
assertTrue(vector1.similarity(vector2) == 1);
}
@Test
public void testProduct() {
Vector vector1 = new SparseVector();
vector1.add(10, 3);
vector1.add(20, 4);
System.out.println(vector1);
Vector vector2 = new SparseVector();
vector2.add(10, 1);
vector2.add(20, 2);
System.out.println(vector2);
assertTrue(vector1.product(vector2) == 11);
}
}