/**
* Copyright (C) 2014-2016 LinkedIn Corp. (pinot-core@linkedin.com)
*
* Licensed 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 com.linkedin.pinot.index.reader;
import com.linkedin.pinot.common.segment.ReadMode;
import com.linkedin.pinot.core.io.reader.impl.v2.FixedBitSingleValueReader;
import com.linkedin.pinot.core.io.writer.impl.v2.FixedBitSingleValueWriter;
import com.linkedin.pinot.core.segment.memory.PinotDataBuffer;
import java.io.File;
import java.nio.channels.FileChannel;
import java.util.Arrays;
import java.util.Random;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.testng.annotations.Test;
public class FixedBitSingleValueTest {
private static final Logger LOGGER = LoggerFactory.getLogger(FixedBitSingleValueTest.class);
@Test
public void testV2() throws Exception {
int ROWS = 1000;
for (int numBits = 1; numBits < 32; numBits++) {
File file = new File(this.getClass().getName() + "_" + numBits + ".test");
FixedBitSingleValueWriter writer = new FixedBitSingleValueWriter(file, ROWS, numBits);
int data[] = new int[ROWS];
Random random = new Random();
int max = (int) Math.pow(2, numBits);
for (int i = 0; i < ROWS; i++) {
data[i] = random.nextInt(max);
writer.setInt(i, data[i]);
}
writer.close();
PinotDataBuffer heapBuffer = PinotDataBuffer.fromFile(file, ReadMode.heap, FileChannel.MapMode.READ_ONLY, "testing");
FixedBitSingleValueReader reader = new FixedBitSingleValueReader(heapBuffer, ROWS, numBits);
int[] read = new int[ROWS];
for (int i = 0; i < ROWS; i++) {
read[i] = reader.getInt(i);
//Assert.assertEquals(reader.getInt(i), data[i],
// "Failed for bit:" + numBits + " Expected " + data[i] + " but found " + reader.getInt(i) + " at " + i);
}
LOGGER.trace(Arrays.toString(data));
LOGGER.trace(Arrays.toString(read));
reader.close();
heapBuffer.close();
file.delete();
}
}
}