@Override public Object getWritableConstantValue() { if (value==null) { return null; } return new ShortWritable(value); } }
@Override public Object create(short value) { return new ShortWritable(value); }
public LazyShort(LazyShortObjectInspector oi) { super(oi); data = new ShortWritable(); }
LazyBinaryShort(WritableShortObjectInspector oi) { super(oi); data = new ShortWritable(); }
public ShortWritable terminate() { return mEmpty ? null : new ShortWritable(mMin); } }
@Override public Object create(final short val) { return new ShortWritable(val); }
public LazyShort(LazyShort copy) { super(copy); data = new ShortWritable(copy.data.get()); }
public ShortWritable readShort(ShortWritable sw) throws IOException { if (sw == null) { sw = new ShortWritable(); } sw.set(in.readShort()); return sw; }
protected List<Object> getFilteredValue(byte alias, Object row) throws HiveException { boolean hasFilter = hasFilter(alias); List<Object> nr = JoinUtil.computeValues(row, joinValues[alias], joinValuesObjectInspectors[alias], hasFilter); if (hasFilter) { short filterTag = JoinUtil.isFiltered(row, joinFilters[alias], joinFilterObjectInspectors[alias], filterMaps[alias]); nr.add(new ShortWritable(filterTag)); aliasFilterTags[alias] &= filterTag; } return nr; }
@Test public void testSet() { ShortWritable obj = new ShortWritable(); assertEquals(new ShortWritable((short) 12), inspector.set(obj, (short) 12)); }
private void runAndVerify(String str, short months, String expResult, GenericUDF udf) throws HiveException { DeferredObject valueObj0 = new DeferredJavaObject(new Text(str)); DeferredObject valueObj1 = new DeferredJavaObject(new ShortWritable(months)); DeferredObject[] args = { valueObj0, valueObj1 }; Text output = (Text) udf.evaluate(args); assertEquals("add_months() test ", expResult, output != null ? output.toString() : null); }
@Test public void testSerialization() throws Exception { MapJoinRowContainer container1 = new MapJoinEagerRowContainer(); container1.addRow(new Object[]{ new Text("f0"), null, new ShortWritable((short)0xf)}); container1.addRow(Arrays.asList(new Object[]{ null, new Text("f1"), new ShortWritable((short)0xf)})); container1.addRow(new Object[]{ null, null, new ShortWritable((short)0xf)}); container1.addRow(Arrays.asList(new Object[]{ new Text("f0"), new Text("f1"), new ShortWritable((short)0x1)})); MapJoinRowContainer container2 = Utilities.serde(container1, "f0,f1,filter", "string,string,smallint"); Utilities.testEquality(container1, container2); Assert.assertEquals(4, container1.rowCount()); Assert.assertEquals(1, container2.getAliasFilter()); }
@Test public void testRegularMap() { final Writable[] entry1 = new Writable[]{new IntWritable(0), new IntWritable(1)}; final Writable[] entry2 = new Writable[]{new IntWritable(2), new IntWritable(3)}; final ArrayWritable map = new ArrayWritable(ArrayWritable.class, new Writable[]{ new ArrayWritable(Writable.class, entry1), new ArrayWritable(Writable.class, entry2)}); assertEquals("Wrong result of inspection", new IntWritable(1), inspector.getMapValueElement(map, new IntWritable(0))); assertEquals("Wrong result of inspection", new IntWritable(3), inspector.getMapValueElement(map, new IntWritable(2))); assertEquals("Wrong result of inspection", new IntWritable(1), inspector.getMapValueElement(map, new ShortWritable((short) 0))); assertEquals("Wrong result of inspection", new IntWritable(3), inspector.getMapValueElement(map, new ShortWritable((short) 2))); }
@Test public void testGet() { ShortWritable obj = new ShortWritable((short) 15); assertEquals((short) 15, inspector.get(obj)); } }
@Test public void testRegularMap() { final Writable[] entry1 = new Writable[]{new IntWritable(0), new IntWritable(1)}; final Writable[] entry2 = new Writable[]{new IntWritable(2), new IntWritable(3)}; final ArrayWritable map = new ArrayWritable(ArrayWritable.class, new Writable[]{ new ArrayWritable(Writable.class, entry1), new ArrayWritable(Writable.class, entry2)}); assertEquals("Wrong result of inspection", new IntWritable(1), inspector.getMapValueElement(map, new IntWritable(0))); assertEquals("Wrong result of inspection", new IntWritable(3), inspector.getMapValueElement(map, new IntWritable(2))); assertNull("Wrong result of inspection", inspector.getMapValueElement(map, new ShortWritable((short) 0))); assertNull("Wrong result of inspection", inspector.getMapValueElement(map, new ShortWritable((short) 2))); }
@Test public void testCreate() { assertEquals(new ShortWritable((short) 8), inspector.create((short) 8)); }
@Test public void testIntWritable() { IntWritable obj = new IntWritable(10); assertEquals(new ShortWritable((short) 10), inspector.getPrimitiveWritableObject(obj)); assertEquals((short) 10, inspector.getPrimitiveJavaObject(obj)); }
@Test public void testEmptyContainer() { final ArrayWritable map = new ArrayWritable(ArrayWritable.class, new ArrayWritable[0]); assertNull("Should be null", inspector.getMapValueElement(map, new ShortWritable((short) 0))); }
@Test public void testNullContainer() { final ArrayWritable map = new ArrayWritable(ArrayWritable.class, null); assertNull("Should be null", inspector.getMapValueElement(map, new ShortWritable((short) 0))); }
@Test public void testShortWritable() { ShortWritable obj = new ShortWritable((short) 5); assertEquals(obj, inspector.getPrimitiveWritableObject(obj)); assertEquals((short) 5, inspector.getPrimitiveJavaObject(obj)); }