/**
* 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.
*/
package net.myrrix.common.collection;
import java.util.Map;
import com.google.common.collect.Maps;
import org.apache.commons.math3.random.RandomGenerator;
import org.junit.Test;
import net.myrrix.common.MyrrixTest;
import net.myrrix.common.random.RandomManager;
public final class FastByIDFloatMapTest extends MyrrixTest {
@Test
public void testPutAndGet() {
FastByIDFloatMap map = new FastByIDFloatMap();
assertNaN(map.get(500000L));
map.put(500000L, 2.0f);
assertEquals(2.0f, map.get(500000L));
}
@Test
public void testRemove() {
FastByIDFloatMap map = new FastByIDFloatMap();
map.put(500000L, 2.0f);
map.remove(500000L);
assertEquals(0, map.size());
assertTrue(map.isEmpty());
assertNaN(map.get(500000L));
}
@Test
public void testClear() {
FastByIDFloatMap map = new FastByIDFloatMap();
map.put(500000L, 2.0f);
map.clear();
assertEquals(0, map.size());
assertTrue(map.isEmpty());
assertNaN(map.get(500000L));
}
@Test
public void testSizeEmpty() {
FastByIDFloatMap map = new FastByIDFloatMap();
assertEquals(0, map.size());
assertTrue(map.isEmpty());
map.put(500000L, 2.0f);
assertEquals(1, map.size());
assertFalse(map.isEmpty());
map.remove(500000L);
assertEquals(0, map.size());
assertTrue(map.isEmpty());
}
@Test
public void testContains() {
FastByIDFloatMap map = buildTestFastMap();
assertTrue(map.containsKey(500000L));
assertTrue(map.containsKey(47L));
assertTrue(map.containsKey(2L));
assertFalse(map.containsKey(999));
}
@Test
public void testRehash() {
FastByIDFloatMap map = buildTestFastMap();
map.remove(500000L);
map.rehash();
assertNaN(map.get(500000L));
assertEquals(3.0f, map.get(47L));
}
@Test
public void testVersusHashMap() {
FastByIDFloatMap actual = new FastByIDFloatMap();
Map<Long,Float> expected = Maps.newHashMapWithExpectedSize(1000000);
RandomGenerator r = RandomManager.getRandom();
for (int i = 0; i < 1000000; i++) {
double d = r.nextDouble();
Long key = (long) r.nextInt(100);
if (d < 0.4) {
Number expectedValue = expected.get(key);
float actualValue = actual.get(key);
if (expectedValue == null) {
assertNaN(actualValue);
} else {
assertEquals(expectedValue.floatValue(), actualValue);
}
} else {
if (d < 0.7) {
expected.put(key, 3.0f);
actual.put(key, 3.0f);
} else {
expected.remove(key);
actual.remove(key);
}
assertEquals(expected.size(), actual.size());
assertEquals(expected.isEmpty(), actual.isEmpty());
}
}
}
@Test
public void testMaxSize() {
FastByIDFloatMap map = new FastByIDFloatMap();
map.put(4, 3.0f);
assertEquals(1, map.size());
map.put(47L, 3.0f);
assertEquals(2, map.size());
assertNaN(map.get(500000L));
map.put(47L, 5.0f);
assertEquals(2, map.size());
assertEquals(5.0f, map.get(47L));
}
private static FastByIDFloatMap buildTestFastMap() {
FastByIDFloatMap map = new FastByIDFloatMap();
map.put(500000L, 2.0f);
map.put(47L, 3.0f);
map.put(2L, 5.0f);
return map;
}
}