org.apache.drill.test.rowSet

Class RowSetUtilities

java.lang.Object

org.apache.drill.test.rowSet.RowSetUtilities

public class RowSetUtilities
extends Object

Various utilities useful for working with row sets, especially for testing.

Method Summary

Modifier and Type	Method and Description
static void	assertEqualValues(String msg, ValueType type, Object expectedObj, Object actualObj)
static void	assertEqualValues(ValueType type, Object expectedObj, Object actualObj)
static void	assertSchemasEqual(TupleMetadata expected, BatchSchema actual)
static byte[][]	binArray(byte[]... elements)
static boolean[]	boolArray(Boolean... elements)
static byte[]	byteArray(Integer... elements)
static BigDecimal	dec(String value)
static BigDecimal[]	decArray(BigDecimal... elements)
static double[]	doubleArray(Double... elements)
static int[]	intArray(Integer... elements)
static Object[]	listValue(Object... elements)
static Long[]	longArray(Long... elements)
static Map<Object,Object>	map(Object... entry) Bootstrap a map object given key-value sequence.
static Object[][]	mapArray(Object[]... elements)
static Object[]	mapValue(Object... members)
static Object[]	objArray(Object... elements)
static org.joda.time.Period	periodFromInt(TypeProtos.MinorType minorType, int value) Ad-hoc, test-only method to set a Period from an integer.
static void	reverse(SelectionVector2 sv2) Reverse a row set by reversing the entries in an SV2.
static void	setFromInt(RowSetWriter rowWriter, int index, int value) Set a test data value from an int.
static Object[]	singleList(Object element)
static Object[]	singleMap(Object member)
static Object[]	singleObjArray(Object element)
static String[]	strArray(String... elements)
static Object	testDataFromInt(ValueType valueType, TypeProtos.MajorType dataType, int value) Create a test value that can be passed to setObject().
static Object[]	variantArray(Object... elements)
static void	verify(RowSet expected, RowSet actual) Verify the actual results, then free memory for both the expected and actual result sets.
static void	verify(RowSet expected, RowSet actual, int rowCount)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

reverse

public static void reverse(SelectionVector2 sv2)

Reverse a row set by reversing the entries in an SV2. This is a quick and easy way to reverse the sort order of an expected-value row set.

Parameters:

sv2 - the SV2 which is reversed in place

setFromInt

public static void setFromInt(RowSetWriter rowWriter,
                              int index,
                              int value)

Set a test data value from an int. Uses the type information of the column to handle interval types. Else, uses the value type of the accessor. The value set here is purely for testing; the mapping from ints to intervals has no real meaning.

Parameters:

rowWriter - writer where value will be written to

index - target index

value - value to write

testDataFromInt

public static Object testDataFromInt(ValueType valueType,
                                     TypeProtos.MajorType dataType,
                                     int value)

Create a test value that can be passed to setObject(). This value matches the value type for a writer.

periodFromInt

public static org.joda.time.Period periodFromInt(TypeProtos.MinorType minorType,
                                                 int value)

Ad-hoc, test-only method to set a Period from an integer. Periods are made up of months and millseconds. There is no mapping from one to the other, so a period requires at least two number. Still, we are given just one (typically from a test data generator.) Use that int value to "spread" some value across the two kinds of fields. The result has no meaning, but has the same comparison order as the original ints.

Parameters:

minorType - the Drill data type

value - the integer value to apply

assertEqualValues

public static void assertEqualValues(ValueType type,
                                     Object expectedObj,
                                     Object actualObj)

assertEqualValues

public static void assertEqualValues(String msg,
                                     ValueType type,
                                     Object expectedObj,
                                     Object actualObj)

mapValue

public static Object[] mapValue(Object... members)

singleMap

public static Object[] singleMap(Object member)

byteArray

public static byte[] byteArray(Integer... elements)

longArray

public static Long[] longArray(Long... elements)

doubleArray

public static double[] doubleArray(Double... elements)

boolArray

public static boolean[] boolArray(Boolean... elements)

strArray

public static String[] strArray(String... elements)

decArray

public static BigDecimal[] decArray(BigDecimal... elements)

binArray

public static byte[][] binArray(byte[]... elements)

intArray

public static int[] intArray(Integer... elements)

mapArray

public static Object[][] mapArray(Object[]... elements)

variantArray

public static Object[] variantArray(Object... elements)

listValue

public static Object[] listValue(Object... elements)

singleList

public static Object[] singleList(Object element)

objArray

public static Object[] objArray(Object... elements)

singleObjArray

public static Object[] singleObjArray(Object element)

verify

public static void verify(RowSet expected,
                          RowSet actual)

Verify the actual results, then free memory for both the expected and actual result sets.

Parameters:

expected - The expected results.

actual - the actual results to verify.

verify

public static void verify(RowSet expected,
                          RowSet actual,
                          int rowCount)

dec

public static BigDecimal dec(String value)

map

public static Map<Object,Object> map(Object... entry)

Bootstrap a map object given key-value sequence.

Parameters:

entry - key-value sequence

Returns:

map containing key-value pairs from passed sequence

assertSchemasEqual

public static void assertSchemasEqual(TupleMetadata expected,
                                      BatchSchema actual)