package org.apache.cassandra.utils; /* * * 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. * */ import org.junit.Test; import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.io.*; import static org.junit.Assert.*; import org.apache.cassandra.db.TypeSizes; import org.apache.cassandra.io.ISerializer; import org.apache.cassandra.io.IVersionedSerializer; public class IntervalTreeTest { @Test public void testSearch() throws Exception { List<Interval<Integer, Void>> intervals = new ArrayList<Interval<Integer, Void>>(); intervals.add(Interval.<Integer, Void>create(-300, -200)); intervals.add(Interval.<Integer, Void>create(-3, -2)); intervals.add(Interval.<Integer, Void>create(1,2)); intervals.add(Interval.<Integer, Void>create(3,6)); intervals.add(Interval.<Integer, Void>create(2,4)); intervals.add(Interval.<Integer, Void>create(5,7)); intervals.add(Interval.<Integer, Void>create(1,3)); intervals.add(Interval.<Integer, Void>create(4,6)); intervals.add(Interval.<Integer, Void>create(8,9)); intervals.add(Interval.<Integer, Void>create(15,20)); intervals.add(Interval.<Integer, Void>create(40,50)); intervals.add(Interval.<Integer, Void>create(49,60)); IntervalTree<Integer, Void, Interval<Integer, Void>> it = IntervalTree.build(intervals); assertEquals(3, it.search(Interval.<Integer, Void>create(4,4)).size()); assertEquals(4, it.search(Interval.<Integer, Void>create(4, 5)).size()); assertEquals(7, it.search(Interval.<Integer, Void>create(-1,10)).size()); assertEquals(0, it.search(Interval.<Integer, Void>create(-1,-1)).size()); assertEquals(5, it.search(Interval.<Integer, Void>create(1,4)).size()); assertEquals(2, it.search(Interval.<Integer, Void>create(0,1)).size()); assertEquals(0, it.search(Interval.<Integer, Void>create(10,12)).size()); List<Interval<Integer, Void>> intervals2 = new ArrayList<Interval<Integer, Void>>(); //stravinsky 1880-1971 intervals2.add(Interval.<Integer, Void>create(1880, 1971)); //Schoenberg intervals2.add(Interval.<Integer, Void>create(1874, 1951)); //Grieg intervals2.add(Interval.<Integer, Void>create(1843, 1907)); //Schubert intervals2.add(Interval.<Integer, Void>create(1779, 1828)); //Mozart intervals2.add(Interval.<Integer, Void>create(1756, 1828)); //Schuetz intervals2.add(Interval.<Integer, Void>create(1585, 1672)); IntervalTree<Integer, Void, Interval<Integer, Void>> it2 = IntervalTree.build(intervals2); assertEquals(0, it2.search(Interval.<Integer, Void>create(1829, 1842)).size()); List<Void> intersection1 = it2.search(Interval.<Integer, Void>create(1907, 1907)); assertEquals(3, intersection1.size()); intersection1 = it2.search(Interval.<Integer, Void>create(1780, 1790)); assertEquals(2, intersection1.size()); } @Test public void testIteration() { List<Interval<Integer, Void>> intervals = new ArrayList<Interval<Integer, Void>>(); intervals.add(Interval.<Integer, Void>create(-300, -200)); intervals.add(Interval.<Integer, Void>create(-3, -2)); intervals.add(Interval.<Integer, Void>create(1,2)); intervals.add(Interval.<Integer, Void>create(3,6)); intervals.add(Interval.<Integer, Void>create(2,4)); intervals.add(Interval.<Integer, Void>create(5,7)); intervals.add(Interval.<Integer, Void>create(1,3)); intervals.add(Interval.<Integer, Void>create(4,6)); intervals.add(Interval.<Integer, Void>create(8,9)); intervals.add(Interval.<Integer, Void>create(15,20)); intervals.add(Interval.<Integer, Void>create(40,50)); intervals.add(Interval.<Integer, Void>create(49,60)); IntervalTree<Integer, Void, Interval<Integer, Void>> it = IntervalTree.build(intervals); Collections.sort(intervals, it.minOrdering); List<Interval<Integer, Void>> l = new ArrayList<Interval<Integer, Void>>(); for (Interval<Integer, Void> i : it) l.add(i); assertEquals(intervals, l); } @Test public void testSerialization() throws Exception { List<Interval<Integer, String>> intervals = new ArrayList<Interval<Integer, String>>(); intervals.add(Interval.<Integer, String>create(-300, -200, "a")); intervals.add(Interval.<Integer, String>create(-3, -2, "b")); intervals.add(Interval.<Integer, String>create(1,2, "c")); intervals.add(Interval.<Integer, String>create(1,3, "d")); intervals.add(Interval.<Integer, String>create(2,4, "e")); intervals.add(Interval.<Integer, String>create(3,6, "f")); intervals.add(Interval.<Integer, String>create(4,6, "g")); intervals.add(Interval.<Integer, String>create(5,7, "h")); intervals.add(Interval.<Integer, String>create(8,9, "i")); intervals.add(Interval.<Integer, String>create(15,20, "j")); intervals.add(Interval.<Integer, String>create(40,50, "k")); intervals.add(Interval.<Integer, String>create(49,60, "l")); IntervalTree<Integer, String, Interval<Integer, String>> it = IntervalTree.build(intervals); IVersionedSerializer<IntervalTree<Integer, String, Interval<Integer, String>>> serializer = IntervalTree.serializer( new ISerializer<Integer>() { public void serialize(Integer i, DataOutput out) throws IOException { out.writeInt(i); } public Integer deserialize(DataInput in) throws IOException { return in.readInt(); } public long serializedSize(Integer i, TypeSizes ts) { return 4; } }, new ISerializer<String>() { public void serialize(String v, DataOutput out) throws IOException { out.writeUTF(v); } public String deserialize(DataInput in) throws IOException { return in.readUTF(); } public long serializedSize(String v, TypeSizes ts) { return v.length(); } }, Interval.class.getConstructor(Object.class, Object.class, Object.class) ); ByteArrayOutputStream baos = new ByteArrayOutputStream(); DataOutputStream out = new DataOutputStream(baos); serializer.serialize(it, out, 0); DataInputStream in = new DataInputStream(new ByteArrayInputStream(baos.toByteArray())); IntervalTree<Integer, String, Interval<Integer, String>> it2 = serializer.deserialize(in, 0); List<Interval<Integer, String>> intervals2 = new ArrayList<Interval<Integer, String>>(); for (Interval<Integer, String> i : it2) intervals2.add(i); assertEquals(intervals, intervals2); } }