objecttocountermaptest.java

一个自然语言处理的Java开源工具包。LingPipe目前已有很丰富的功能

package com.aliasi.test.unit.util;

import com.aliasi.util.ObjectToCounterMap;

import com.aliasi.test.unit.BaseTestCase;

import java.util.Arrays;
import java.util.List;

public class ObjectToCounterMapTest extends BaseTestCase {

    public void testConstructor() {
        ObjectToCounterMap map = new ObjectToCounterMap();
        assertNotNull(map);
    }

    public void testIncrement() {
        ObjectToCounterMap map = new ObjectToCounterMap();
    assertEquals(0,map.getCount("a"));
        map.increment("a");
        assertEquals(1,map.getCount("a"));
        map.increment("a");
        assertEquals(2,map.getCount("a"));
        map.increment("a",5);
        assertEquals(7,map.getCount("a"));
        assertTrue(map.containsKey("a"));
        map.increment("a",-7);
        assertFalse(map.containsKey("a"));
    }

    public void testSet() {
        ObjectToCounterMap map = new ObjectToCounterMap();
        map.set("a",3);
        assertEquals(3,map.getCount("a"));
        assertTrue(map.containsKey("a"));
        map.set("a",0);
        assertEquals(0,map.getCount("a"));
        assertFalse(map.containsKey("a"));
        map.set("a",3);
        map.set("a",4);
        assertEquals(4,map.getCount("a"));
        map.set("b",17);
        assertEquals(17,map.getCount("b"));
    }

    public void testKeysOrderedByCount() {
        ObjectToCounterMap map = new ObjectToCounterMap();
        map.set("e",1);
        map.set("c",3);
        map.set("d",2);
        map.set("a",5);
        List keysOrderedByCount = map.keysOrderedByCountList();
        assertEquals(Arrays.asList(new Object[] { "a", "c", "d", "e" }),
                          keysOrderedByCount);
    }

    public void testCountComparator() {
        // two incomparables
        ObjectToCounterMap map = new ObjectToCounterMap();
        Object o1 = new Object();
        Object o2 = new Object();
        map.set(o1,2);
        map.set(o2,2);
        assertEquals(0,map.countComparator().compare(o1,o2));
        map.set(o1,3);
        assertEquals(-1,map.countComparator().compare(o1,o2));
        map.set(o1,1);
        assertEquals(1,map.countComparator().compare(o1,o2));

        // two comparaombles
        String s1 = "a";
        String s2 = "b";
        map.set(s1,2);
        map.set(s2,2);
        assertEquals(-1,map.countComparator().compare(s1,s2));
        map.set(s1,3);
        assertEquals(-1,map.countComparator().compare(s1,s2));
        map.set(s1,1);
        assertEquals(1,map.countComparator().compare(s1,s2));
        map.set(o1,1);
        map.set(s1,1);
        assertEquals(0,map.countComparator().compare(o1,s1));
    }

    public void testPrune() {
    ObjectToCounterMap map = new ObjectToCounterMap();
    Object o1 = new Integer(1);
    Object o2 = new Integer(2);
    Object o3 = new Integer(3);
    map.set(o1,1);
    map.set(o2,1);
    map.set(o3,3);
    assertEquals(3,map.size());
    assertEquals(3,map.getCount(o3));
    assertEquals(1,map.getCount(o1));
    assertEquals(0,map.getCount(new Integer(10)));
    map.prune(1);
    assertEquals(3,map.size());
    map.prune(2);
    assertEquals(1,map.size());
    assertEquals(3,map.getCount(o3));
    assertEquals(0,map.getCount(o1));
    }

}