test.it.unimi.dsi.big.util.MappedFrontCodedStringBigListTest Maven / Gradle / Ivy

Go to download
/*
 * DSI utilities
 *
 * Copyright (C) 2010-2023 Sebastiano Vigna
 *
 * This program and the accompanying materials are made available under the
 * terms of the GNU Lesser General Public License v2.1 or later,
 * which is available at
 * http://www.gnu.org/licenses/old-licenses/lgpl-2.1-standalone.html,
 * or the Apache Software License 2.0, which is available at
 * https://www.apache.org/licenses/LICENSE-2.0.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * SPDX-License-Identifier: LGPL-2.1-or-later OR Apache-2.0
 */

package it.unimi.dsi.big.util;

import static org.junit.Assert.assertEquals;

import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.apache.commons.configuration2.ex.ConfigurationException;
import org.apache.commons.lang3.StringUtils;
import org.junit.Test;

import it.unimi.dsi.lang.MutableString;

public class MappedFrontCodedStringBigListTest {

	@Test
	public void test() throws IOException, ConfigurationException {
		final String basename = File.createTempFile(this.getClass().getName(), ".basename").toString();
		final List c = new ArrayList<>(Arrays.asList(TernaryIntervalSearchTreeTest.WORDS.clone()));
		c.add(StringUtils.repeat("a", 1000));
		c.add(StringUtils.repeat("a", 500) + StringUtils.repeat("b", 500));
		c.add(StringUtils.repeat("a", 1000) + StringUtils.repeat("b", 1000));
		c.add(StringUtils.repeat("a", 100) + StringUtils.repeat("b", 1000));
		final MutableString s = new MutableString();
		Collections.sort(c);
		for (int p = 0; p < 2; p++) {
			for (int ratio = 1; ratio < 8; ratio++) {
				final FrontCodedStringBigList fcl = new FrontCodedStringBigList(c.iterator(), ratio, true);

				MappedFrontCodedStringBigList.build(basename, 4, c.stream().map(x -> x.getBytes(StandardCharsets.UTF_8)).iterator());
				final MappedFrontCodedStringBigList mfcl = MappedFrontCodedStringBigList.load(basename);
				for (int i = 0; i < fcl.size64(); i++) {
					assertEquals(Integer.toString(i), c.get(i), mfcl.get(i).toString());
					assertEquals(Integer.toString(i), c.get(i), mfcl.getString(i));
					assertEquals(Integer.toString(i), c.get(i), new String(mfcl.getArray(i), StandardCharsets.UTF_8));
					fcl.get(i, s);
					assertEquals(Integer.toString(i), c.get(i), s.toString());
				}
			}
		}

		new File(basename + MappedFrontCodedStringBigList.PROPERTIES_EXTENSION).delete();
		new File(basename + MappedFrontCodedStringBigList.BYTE_ARRAY_EXTENSION).delete();
		new File(basename + MappedFrontCodedStringBigList.POINTERS_EXTENSION).delete();
	}
}