def test_TopN1(self): counterHashMap = CounterHashMap() counterHashMap.put("item1") counterHashMap.put("item2") counterHashMap.put("item3") counterHashMap.put("item1") counterHashMap.put("item2") counterHashMap.put("item1") self.assertEquals("item1", counterHashMap.topN(1)[0][1]) self.assertEquals("item2", counterHashMap.topN(2)[1][1]) self.assertEquals("item3", counterHashMap.topN(3)[2][1])
def test_TransitionWithName(self): transitionCounts = CounterHashMap() for state in self.stateList: transitions = self.fsm.getTransitions(state) for transition in transitions: transitionCounts.put(transition.withName()) topList = transitionCounts.topN(5) self.assertEqual(None, topList[0][1]) self.assertEqual(52, topList[0][0]) self.assertEqual("^DB+VERB+CAUS", topList[1][1]) self.assertEqual(33, topList[1][0]) self.assertEqual("^DB+VERB+PASS", topList[2][1]) self.assertEqual(31, topList[2][0]) self.assertEqual("A3PL", topList[3][1]) self.assertEqual(28, topList[3][0]) self.assertEqual("LOC", topList[4][1]) self.assertEqual(24, topList[4][0])
def test_TransitionWith(self): transitionCounts = CounterHashMap() for state in self.stateList: transitions = self.fsm.getTransitions(state) for transition in transitions: transitionCounts.put(transition.__str__()) topList = transitionCounts.topN(5) self.assertEqual("0", topList[0][1]) self.assertEqual(111, topList[0][0]) self.assertEqual("lAr", topList[1][1]) self.assertEqual(37, topList[1][0]) self.assertEqual("DHr", topList[2][1]) self.assertEqual(28, topList[2][0]) self.assertEqual("Hn", topList[3][1]) self.assertEqual(24, topList[3][0]) self.assertEqual("lArH", topList[4][1]) self.assertEqual(23, topList[4][0])
def test_TopN2(self): counterHashMap = CounterHashMap() for i in range(1000): counterHashMap.putNTimes(i, 2 * i + 2) self.assertEquals(990, counterHashMap.topN(10)[9][1]) self.assertEquals(900, counterHashMap.topN(100)[99][1])