package org.apache.lucene.util.collections;
import org.junit.Test;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Random;
import org.apache.lucene.util.LuceneTestCase;
import org.apache.lucene.util.collections.FloatIterator;
import org.apache.lucene.util.collections.FloatToObjectMap;
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
public class FloatToObjectMapTest extends LuceneTestCase {
@Test
public void test0() {
FloatToObjectMap<Integer> map = new FloatToObjectMap<Integer>();
assertNull(map.get(0));
for (int i = 0; i < 100; ++i) {
int value = 100 + i;
assertFalse(map.containsValue(value));
map.put(i * 1.1f, value);
assertTrue(map.containsValue(value));
assertNotNull(map.get(i * 1.1f));
}
assertEquals(100, map.size());
for (int i = 0; i < 100; ++i) {
assertTrue(map.containsKey(i*1.1f));
assertEquals(100 + i, map.get(i*1.1f).intValue());
}
for (int i = 10; i < 90; ++i) {
map.remove(i*1.1f);
assertNull(map.get(i*1.1f));
}
assertEquals(20, map.size());
for (int i = 0; i < 100; ++i) {
assertEquals(map.containsKey(i*1.1f), !(i >= 10 && i < 90));
}
for (int i = 5; i < 85; ++i) {
map.put(i*1.1f, Integer.valueOf(5 + i));
}
assertEquals(95, map.size());
for (int i = 0; i < 100; ++i) {
assertEquals(map.containsKey(i*1.1f), !(i >= 85 && i < 90));
}
for (int i = 0; i < 5; ++i) {
assertEquals(map.get(i*1.1f).intValue(), (100 + i));
}
for (int i = 5; i < 85; ++i) {
assertEquals(map.get(i*1.1f).intValue(), (5 + i));
}
for (int i = 90; i < 100; ++i) {
assertEquals(map.get(i*1.1f).intValue(), (100 + i));
}
}
@Test
public void test1() {
FloatToObjectMap<Integer> map = new FloatToObjectMap<Integer>();
for (int i = 0; i < 100; ++i) {
map.put(i*1.1f, Integer.valueOf(100 + i));
}
HashSet<Integer> set = new HashSet<Integer>();
for (Iterator<Integer> iterator = map.iterator(); iterator.hasNext();) {
set.add(iterator.next());
}
assertEquals(set.size(), map.size());
for (int i = 0; i < 100; ++i) {
assertTrue(set.contains(Integer.valueOf(100 + i)));
}
set.clear();
for (Iterator<Integer> iterator = map.iterator(); iterator.hasNext();) {
Integer integer = iterator.next();
if (integer % 2 == 1) {
iterator.remove();
continue;
}
set.add(integer);
}
assertEquals(set.size(), map.size());
for (int i = 0; i < 100; i += 2) {
assertTrue(set.contains(Integer.valueOf(100 + i)));
}
}
@Test
public void test2() {
FloatToObjectMap<Integer> map = new FloatToObjectMap<Integer>();
assertTrue(map.isEmpty());
assertNull(map.get(0));
for (int i = 0; i < 128; ++i) {
int value = i * 4096;
assertFalse(map.containsValue(value));
map.put(i*1.1f, value);
assertTrue(map.containsValue(value));
assertNotNull(map.get(i*1.1f));
assertFalse(map.isEmpty());
}
assertEquals(128, map.size());
for (int i = 0; i < 128; ++i) {
assertTrue(map.containsKey(i*1.1f));
assertEquals(i * 4096, map.get(i*1.1f).intValue());
}
for (int i = 0; i < 200; i += 2) {
map.remove(i*1.1f);
}
assertEquals(64, map.size());
for (int i = 1; i < 128; i += 2) {
assertTrue(map.containsKey(i*1.1f));
assertEquals(i * 4096, map.get(i*1.1f).intValue());
map.remove(i*1.1f);
}
assertTrue(map.isEmpty());
}
@Test
public void test3() {
FloatToObjectMap<Integer> map = new FloatToObjectMap<Integer>();
int length = 100;
for (int i = 0; i < length; ++i) {
map.put(i * 64*1.1f, 100 + i);
}
HashSet<Float> keySet = new HashSet<Float>();
for (FloatIterator iit = map.keyIterator(); iit.hasNext();) {
keySet.add(iit.next());
}
assertEquals(length, keySet.size());
for (int i = 0; i < length; ++i) {
assertTrue(keySet.contains(i * 64*1.1f));
}
HashSet<Integer> valueSet = new HashSet<Integer>();
for (Iterator<Integer> iit = map.iterator(); iit.hasNext();) {
valueSet.add(iit.next());
}
assertEquals(length, valueSet.size());
Object[] array = map.toArray();
assertEquals(length, array.length);
for (Object value : array) {
assertTrue(valueSet.contains(value));
}
Integer[] array2 = new Integer[80];
array2 = map.toArray(array2);
for (int value : array2) {
assertTrue(valueSet.contains(value));
}
Integer[] array3 = new Integer[120];
array3 = map.toArray(array3);
for (int i = 0; i < length; ++i) {
assertTrue(valueSet.contains(array3[i]));
}
assertNull(array3[length]);
for (int i = 0; i < length; ++i) {
assertTrue(map.containsValue(i + 100));
assertTrue(map.containsKey(i * 64*1.1f));
}
for (FloatIterator iit = map.keyIterator(); iit.hasNext();) {
iit.next();
iit.remove();
}
assertTrue(map.isEmpty());
assertEquals(0, map.size());
}
// now with random data.. and lots of it
@Test
public void test4() {
FloatToObjectMap<Integer> map = new FloatToObjectMap<Integer>();
int length = ArrayHashMapTest.RANDOM_TEST_NUM_ITERATIONS;
// for a repeatable random sequence
long seed = random().nextLong();
Random random = new Random(seed);
for (int i = 0; i < length; ++i) {
int value = random.nextInt(Integer.MAX_VALUE);
map.put(i * 128*1.1f, value);
}
assertEquals(length, map.size());
// now repeat
random.setSeed(seed);
for (int i = 0; i < length; ++i) {
int value = random.nextInt(Integer.MAX_VALUE);
assertTrue(map.containsValue(value));
assertTrue(map.containsKey(i * 128*1.1f));
assertEquals(Integer.valueOf(value), map.remove(i * 128*1.1f));
}
assertEquals(0, map.size());
assertTrue(map.isEmpty());
}
@Test
public void testEquals() {
FloatToObjectMap<Integer> map1 = new FloatToObjectMap<Integer>(100);
FloatToObjectMap<Integer> map2 = new FloatToObjectMap<Integer>(100);
assertEquals("Empty maps should be equal", map1, map2);
assertEquals("hashCode() for empty maps should be equal",
map1.hashCode(), map2.hashCode());
for (int i = 0; i < 100; ++i) {
map1.put(i * 1.1f, 100 + i);
map2.put(i * 1.1f, 100 + i);
}
assertEquals("Identical maps should be equal", map1, map2);
assertEquals("hashCode() for identical maps should be equal",
map1.hashCode(), map2.hashCode());
for (int i = 10; i < 20; i++) {
map1.remove(i * 1.1f);
}
assertFalse("Different maps should not be equal", map1.equals(map2));
for (int i = 19; i >=10; --i) {
map2.remove(i * 1.1f);
}
assertEquals("Identical maps should be equal", map1, map2);
assertEquals("hashCode() for identical maps should be equal",
map1.hashCode(), map2.hashCode());
map1.put(-1.1f,-1);
map2.put(-1.1f,-2);
assertFalse("Different maps should not be equal", map1.equals(map2));
map2.put(-1.1f,-1);
assertEquals("Identical maps should be equal", map1, map2);
assertEquals("hashCode() for identical maps should be equal",
map1.hashCode(), map2.hashCode());
}
}