提交 dc729048 编写于 作者: Z zentol 提交者: StephanEwen

TuplePairComparatorTests

上级 f90f172c
/***********************************************************************************************************************
*
* Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
*
* 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 eu.stratosphere.api.java.typeutils.runtime;
import eu.stratosphere.api.common.typeutils.TypeComparator;
import eu.stratosphere.api.common.typeutils.base.DoubleComparator;
import eu.stratosphere.api.common.typeutils.base.IntComparator;
import eu.stratosphere.api.java.tuple.Tuple2;
import eu.stratosphere.api.java.tuple.Tuple3;
import eu.stratosphere.api.java.typeutils.runtime.tuple.base.TuplePairComparatorTestBase;
public class TuplePairComparatorTest extends TuplePairComparatorTestBase<Tuple3<Integer, String, Double>, Tuple3<Integer, Double, Long>> {
@SuppressWarnings("unchecked")
Tuple3<Integer, String, Double>[] dataISD = new Tuple3[]{
new Tuple3<Integer, String, Double>(4, "hello", 20.0),
new Tuple3<Integer, String, Double>(4, "world", 23.2),
new Tuple3<Integer, String, Double>(5, "hello", 20.0),
new Tuple3<Integer, String, Double>(5, "world", 23.2),
new Tuple3<Integer, String, Double>(6, "hello", 20.0),
new Tuple3<Integer, String, Double>(6, "world", 23.2),
new Tuple3<Integer, String, Double>(7, "hello", 20.0),
new Tuple3<Integer, String, Double>(7, "world", 23.2)
};
@SuppressWarnings("unchecked")
Tuple3<Integer, Double, Long>[] dataIDL = new Tuple3[]{
new Tuple3<Integer, Double, Long>(4, 20.0, new Long(14)),
new Tuple3<Integer, Double, Long>(4, 23.2, new Long(15)),
new Tuple3<Integer, Double, Long>(5, 20.0, new Long(15)),
new Tuple3<Integer, Double, Long>(5, 23.2, new Long(20)),
new Tuple3<Integer, Double, Long>(6, 20.0, new Long(20)),
new Tuple3<Integer, Double, Long>(6, 23.2, new Long(29)),
new Tuple3<Integer, Double, Long>(7, 20.0, new Long(29)),
new Tuple3<Integer, Double, Long>(7, 23.2, new Long(34))
};
@Override
protected TuplePairComparator<Tuple3<Integer, String, Double>, Tuple3<Integer, Double, Long>> createComparator(boolean ascending) {
return new TuplePairComparator<Tuple3<Integer, String, Double>, Tuple3<Integer, Double, Long>>(
new int[]{0, 2},
new int[]{0, 1},
new TypeComparator[]{
new IntComparator(ascending),
new DoubleComparator(ascending)
},
new TypeComparator[]{
new IntComparator(ascending),
new DoubleComparator(ascending)
}
);
}
@Override
protected Tuple2<Tuple3<Integer, String, Double>[], Tuple3<Integer, Double, Long>[]> getSortedTestData() {
return new Tuple2(dataISD, dataIDL);
}
}
/***********************************************************************************************************************
*
* Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
*
* 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 eu.stratosphere.api.java.typeutils.runtime.tuple.base;
import eu.stratosphere.api.java.tuple.Tuple;
import eu.stratosphere.api.java.tuple.Tuple2;
import eu.stratosphere.api.java.typeutils.runtime.TuplePairComparator;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import org.junit.Test;
/**
* Abstract test base for TuplePairComparators.
*
* @param <T>
* @param <R>
*/
public abstract class TuplePairComparatorTestBase<T extends Tuple, R extends Tuple> {
protected abstract TuplePairComparator<T, R> createComparator(boolean ascending);
protected abstract Tuple2<T[], R[]> getSortedTestData();
@Test
public void testEqualityWithReference() {
try {
TuplePairComparator<T, R> comparator = getComparator(true);
Tuple2<T[], R[]> data = getSortedData();
for (int x = 0; x < data.f0.length; x++) {
comparator.setReference(data.f0[x]);
assertTrue(comparator.equalToReference(data.f1[x]));
}
} catch (Exception e) {
System.err.println(e.getMessage());
e.printStackTrace();
fail("Exception in test: " + e.getMessage());
}
}
@Test
public void testInequalityWithReference() {
testGreatSmallAscDescWithReference(true);
testGreatSmallAscDescWithReference(false);
}
protected void testGreatSmallAscDescWithReference(boolean ascending) {
try {
Tuple2<T[], R[]> data = getSortedData();
TuplePairComparator<T, R> comparator = getComparator(ascending);
//compares every element in high with every element in low
for (int x = 0; x < data.f0.length - 1; x++) {
for (int y = x + 1; y < data.f1.length; y++) {
comparator.setReference(data.f0[x]);
if (ascending) {
assertTrue(comparator.compareToReference(data.f1[y]) > 0);
} else {
assertTrue(comparator.compareToReference(data.f1[y]) < 0);
}
}
}
} catch (Exception e) {
System.err.println(e.getMessage());
e.printStackTrace();
fail("Exception in test: " + e.getMessage());
}
}
// --------------------------------------------------------------------------------------------
protected TuplePairComparator<T, R> getComparator(boolean ascending) {
TuplePairComparator<T, R> comparator = createComparator(ascending);
if (comparator == null) {
throw new RuntimeException("Test case corrupt. Returns null as comparator.");
}
return comparator;
}
protected Tuple2<T[], R[]> getSortedData() {
Tuple2<T[], R[]> data = getSortedTestData();
if (data == null || data.f0 == null || data.f1 == null) {
throw new RuntimeException("Test case corrupt. Returns null as test data.");
}
if (data.f0.length < 2 || data.f1.length < 2) {
throw new RuntimeException("Test case does not provide enough sorted test data.");
}
return data;
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册