Search in sources :

Example 16 with MutableBitmap

use of io.druid.collections.bitmap.MutableBitmap in project druid by druid-io.

the class TestIntegerSet method testContainsAll.

@Test
public void testContainsAll() throws IllegalAccessException, InstantiationException {
    for (Class<? extends MutableBitmap> clazz : clazzes) {
        MutableBitmap wrappedBitmap = clazz.newInstance();
        IntSetTestUtility.addAllToMutable(wrappedBitmap, IntSetTestUtility.getSetBits());
        IntegerSet integerSet = IntegerSet.wrap(wrappedBitmap);
        Set<Integer> set = IntSetTestUtility.getSetBits();
        Assert.assertTrue(integerSet.containsAll(set));
        set.add(999);
        Assert.assertFalse(integerSet.containsAll(set));
    }
}
Also used : MutableBitmap(io.druid.collections.bitmap.MutableBitmap) Test(org.junit.Test)

Example 17 with MutableBitmap

use of io.druid.collections.bitmap.MutableBitmap in project druid by druid-io.

the class TestIntegerSet method testToSmallArray.

@Test
public void testToSmallArray() throws IllegalAccessException, InstantiationException {
    for (Class<? extends MutableBitmap> clazz : clazzes) {
        Exception e = null;
        MutableBitmap wrappedBitmap = clazz.newInstance();
        IntSetTestUtility.addAllToMutable(wrappedBitmap, IntSetTestUtility.getSetBits());
        IntegerSet integerSet = IntegerSet.wrap(wrappedBitmap);
        Set<Integer> set = Sets.newHashSet((Integer[]) integerSet.toArray(new Integer[0]));
        Assert.assertTrue(Sets.difference(integerSet, set).isEmpty());
    }
}
Also used : MutableBitmap(io.druid.collections.bitmap.MutableBitmap) Test(org.junit.Test)

Example 18 with MutableBitmap

use of io.druid.collections.bitmap.MutableBitmap in project druid by druid-io.

the class TestIntegerSet method testToBigArray.

@Test
public void testToBigArray() throws IllegalAccessException, InstantiationException {
    for (Class<? extends MutableBitmap> clazz : clazzes) {
        Exception e = null;
        MutableBitmap wrappedBitmap = clazz.newInstance();
        IntSetTestUtility.addAllToMutable(wrappedBitmap, IntSetTestUtility.getSetBits());
        IntegerSet integerSet = IntegerSet.wrap(wrappedBitmap);
        Integer[] bigArray = new Integer[1024];
        integerSet.toArray(bigArray);
        Set<Integer> set = Sets.newHashSet(bigArray);
        Assert.assertTrue(Sets.difference(integerSet, set).isEmpty());
    }
}
Also used : MutableBitmap(io.druid.collections.bitmap.MutableBitmap) Test(org.junit.Test)

Example 19 with MutableBitmap

use of io.druid.collections.bitmap.MutableBitmap in project druid by druid-io.

the class TestIntegerSet method testRemoveOneThing.

@Test
public void testRemoveOneThing() throws IllegalAccessException, InstantiationException {
    for (Class<? extends MutableBitmap> clazz : clazzes) {
        MutableBitmap wrappedBitmap = clazz.newInstance();
        IntSetTestUtility.addAllToMutable(wrappedBitmap, IntSetTestUtility.getSetBits());
        IntegerSet integerSet = IntegerSet.wrap(wrappedBitmap);
        Set<Integer> set = IntSetTestUtility.getSetBits();
        integerSet.remove(1);
        set.remove(1);
        Assert.assertTrue(Sets.difference(set, integerSet).isEmpty());
    }
}
Also used : MutableBitmap(io.druid.collections.bitmap.MutableBitmap) Test(org.junit.Test)

Example 20 with MutableBitmap

use of io.druid.collections.bitmap.MutableBitmap in project druid by druid-io.

the class TestIntegerSet method testIntOverflow.

@Test
public void testIntOverflow() throws IllegalAccessException, InstantiationException {
    for (Class<? extends MutableBitmap> clazz : clazzes) {
        Exception e = null;
        try {
            MutableBitmap wrappedBitmap = clazz.newInstance();
            IntSetTestUtility.addAllToMutable(wrappedBitmap, IntSetTestUtility.getSetBits());
            IntegerSet integerSet = IntegerSet.wrap(wrappedBitmap);
            integerSet.add(Integer.MAX_VALUE + 1);
        } catch (java.lang.IllegalArgumentException ex) {
            e = ex;
        }
        Assert.assertNotNull(e);
    }
}
Also used : MutableBitmap(io.druid.collections.bitmap.MutableBitmap) Test(org.junit.Test)

Aggregations

MutableBitmap (io.druid.collections.bitmap.MutableBitmap)28 Test (org.junit.Test)15 ImmutableBitmap (io.druid.collections.bitmap.ImmutableBitmap)7 BitmapFactory (io.druid.collections.bitmap.BitmapFactory)5 BitmapIndexSelector (io.druid.query.filter.BitmapIndexSelector)4 BitmapIndex (io.druid.segment.column.BitmapIndex)4 BenchmarkOptions (com.carrotsearch.junitbenchmarks.BenchmarkOptions)3 Function (com.google.common.base.Function)3 RoaringBitmapFactory (io.druid.collections.bitmap.RoaringBitmapFactory)3 ImmutableRTree (io.druid.collections.spatial.ImmutableRTree)3 BitmapSerdeFactory (io.druid.segment.data.BitmapSerdeFactory)3 GenericIndexed (io.druid.segment.data.GenericIndexed)3 Indexed (io.druid.segment.data.Indexed)3 RoaringBitmapSerdeFactory (io.druid.segment.data.RoaringBitmapSerdeFactory)3 BitmapIndexColumnPartSupplier (io.druid.segment.serde.BitmapIndexColumnPartSupplier)3 Setup (org.openjdk.jmh.annotations.Setup)3 ConciseBitmapFactory (io.druid.collections.bitmap.ConciseBitmapFactory)1 WrappedImmutableRoaringBitmap (io.druid.collections.bitmap.WrappedImmutableRoaringBitmap)1 WrappedRoaringBitmap (io.druid.collections.bitmap.WrappedRoaringBitmap)1 EmptyIntIterator (io.druid.extendedset.intset.EmptyIntIterator)1