@Override public Object getWritableConstantValue() { if (value==null) { return null; } return new ByteWritable(value); } }
LazyBinaryByte(WritableByteObjectInspector oi) { super(oi); data = new ByteWritable(); }
@Override public Object getPrimitiveWritableObject(Object o) { return o == null ? null : new ByteWritable(((Byte) o).byteValue()); }
LazyBinaryByte(LazyBinaryByte copy) { super(copy); data = new ByteWritable(copy.data.get()); }
@Override public Object copyObject(Object o) { return o == null ? null : new ByteWritable(get(o)); }
public LazyByte(LazyByte copy) { super(copy); data = new ByteWritable(copy.data.get()); }
public ByteWritable readByte(ByteWritable bw) throws IOException { if (bw == null) { bw = new ByteWritable(); } bw.set(in.readByte()); return bw; }
@Test public void testSet() { ByteWritable obj = new ByteWritable(); assertEquals(new ByteWritable((byte) 12), inspector.set(obj, (byte) 12)); }
private void runAndVerify(String str, byte months, String expResult, GenericUDF udf) throws HiveException { DeferredObject valueObj0 = new DeferredJavaObject(new Text(str)); DeferredObject valueObj1 = new DeferredJavaObject(new ByteWritable(months)); DeferredObject[] args = { valueObj0, valueObj1 }; Text output = (Text) udf.evaluate(args); assertEquals("add_months() test ", expResult, output != null ? output.toString() : null); } }
@Test(expected = UDFArgumentException.class) public void testConversionIsPrevented1() throws HiveException { GenericUDFNullif udf = new GenericUDFNullif(); ObjectInspector[] inputOIs = { PrimitiveObjectInspectorFactory.writableStringObjectInspector, PrimitiveObjectInspectorFactory.writableByteObjectInspector }; DeferredObject[] args = { new DeferredJavaObject(new Text("4")), new DeferredJavaObject(new ByteWritable((byte) 4)) }; PrimitiveObjectInspector oi = (PrimitiveObjectInspector) udf.initialize(inputOIs); }
@Test(expected = UDFArgumentException.class) public void testConversionIsPrevented2() throws HiveException { GenericUDFNullif udf = new GenericUDFNullif(); ObjectInspector[] inputOIs = { PrimitiveObjectInspectorFactory.writableByteObjectInspector, PrimitiveObjectInspectorFactory.writableStringObjectInspector }; DeferredObject[] args = { new DeferredJavaObject(new ByteWritable((byte) 4)), new DeferredJavaObject(new Text("4")) }; PrimitiveObjectInspector oi = (PrimitiveObjectInspector) udf.initialize(inputOIs); }
@Test public void testByteTypeEq() throws HiveException { GenericUDFNullif udf = new GenericUDFNullif(); ObjectInspector[] inputOIs = { PrimitiveObjectInspectorFactory.writableByteObjectInspector, PrimitiveObjectInspectorFactory.writableByteObjectInspector }; DeferredObject[] args = { new DeferredJavaObject(new ByteWritable((byte) 4)), new DeferredJavaObject(new ByteWritable((byte) 4)) }; PrimitiveObjectInspector oi = (PrimitiveObjectInspector) udf.initialize(inputOIs); Assert.assertEquals(TypeInfoFactory.byteTypeInfo, oi.getTypeInfo()); ByteWritable res = (ByteWritable) udf.evaluate(args); Assert.assertEquals(null, res); }
@Test public void testByteNeq() throws HiveException { GenericUDFNullif udf = new GenericUDFNullif(); ObjectInspector[] inputOIs = { PrimitiveObjectInspectorFactory.writableByteObjectInspector, PrimitiveObjectInspectorFactory.writableByteObjectInspector }; DeferredObject[] args = { new DeferredJavaObject(new ByteWritable((byte) 4)), new DeferredJavaObject(new ByteWritable((byte) 1)) }; PrimitiveObjectInspector oi = (PrimitiveObjectInspector) udf.initialize(inputOIs); Assert.assertEquals(TypeInfoFactory.byteTypeInfo, oi.getTypeInfo()); ByteWritable res = (ByteWritable) udf.evaluate(args); Assert.assertEquals(4, res.get()); }
public void performSequenceFileRead(FileSystem fs, int count, Path file) throws IOException { SequenceFile.Reader reader = new SequenceFile.Reader(fs, file, conf); ByteWritable key = new ByteWritable(); BytesRefArrayWritable val = new BytesRefArrayWritable(); for (int i = 0; i < count; i++) { reader.next(key, val); } }
@Test public void testGet() { ByteWritable obj = new ByteWritable((byte) 15); assertEquals((byte) 15, inspector.get(obj)); } }
@Test public void testCreate() { assertEquals(new ByteWritable((byte) 8), inspector.create((byte) 8)); }
@Test public void testConversionInSameGroup() throws HiveException { GenericUDFNullif udf = new GenericUDFNullif(); ObjectInspector[] inputOIs = { PrimitiveObjectInspectorFactory.writableDoubleObjectInspector, PrimitiveObjectInspectorFactory.writableByteObjectInspector }; DeferredObject[] args = { new DeferredJavaObject(new DoubleWritable(4.0)), new DeferredJavaObject(new ByteWritable((byte) 4)) }; PrimitiveObjectInspector oi = (PrimitiveObjectInspector) udf.initialize(inputOIs); }
@Test(expected = UDFArgumentException.class) public void testNotSupportedArgumentMix() throws HiveException { GenericUDFNullif udf = new GenericUDFNullif(); ObjectInspector[] inputOIs = { PrimitiveObjectInspectorFactory.writableDateObjectInspector, PrimitiveObjectInspectorFactory.writableByteObjectInspector }; DeferredObject[] args = { new DeferredJavaObject(new DateWritableV2(4)), new DeferredJavaObject(new ByteWritable((byte) 4)) }; udf.initialize(inputOIs); }
@Test public void testIntWritable() { IntWritable obj = new IntWritable(10); assertEquals(new ByteWritable((byte) 10), inspector.getPrimitiveWritableObject(obj)); assertEquals((byte) 10, inspector.getPrimitiveJavaObject(obj)); }
@Test public void testByteWritable() { ByteWritable obj = new ByteWritable((byte) 5); assertEquals(obj, inspector.getPrimitiveWritableObject(obj)); assertEquals((byte) 5, inspector.getPrimitiveJavaObject(obj)); }