private EquatableValueSet checkCompatibility(ValueSet other) { if (!getType().equals(other.getType())) { throw new IllegalStateException(String.format("Mismatched types: %s vs %s", getType(), other.getType())); } if (!(other instanceof EquatableValueSet)) { throw new IllegalStateException(String.format("ValueSet is not a EquatableValueSet: %s", other.getClass())); } return (EquatableValueSet) other; }
public static PrestoThriftEquatableValueSet fromEquatableValueSet(EquatableValueSet valueSet) { Type type = valueSet.getType(); Set<ValueEntry> values = valueSet.getEntries(); List<PrestoThriftBlock> thriftValues = new ArrayList<>(values.size()); for (ValueEntry value : values) { checkState(type.equals(value.getType()), "ValueEntrySet has elements of different types: %s vs %s", type, value.getType()); thriftValues.add(fromBlock(value.getBlock(), type)); } return new PrestoThriftEquatableValueSet(valueSet.isWhiteList(), thriftValues); } }
@Test public void testEntireSet() { EquatableValueSet equatables = EquatableValueSet.all(ID); assertEquals(equatables.getType(), ID); assertFalse(equatables.isNone()); assertTrue(equatables.isAll()); assertFalse(equatables.isSingleValue()); assertFalse(equatables.isWhiteList()); assertEquals(equatables.getValues().size(), 0); assertEquals(equatables.complement(), EquatableValueSet.none(ID)); assertTrue(equatables.containsValue(0L)); assertTrue(equatables.containsValue(1L)); }
@Test public void testEmptySet() { EquatableValueSet equatables = EquatableValueSet.none(ID); assertEquals(equatables.getType(), ID); assertTrue(equatables.isNone()); assertFalse(equatables.isAll()); assertFalse(equatables.isSingleValue()); assertTrue(equatables.isWhiteList()); assertEquals(equatables.getValues().size(), 0); assertEquals(equatables.complement(), EquatableValueSet.all(ID)); assertFalse(equatables.containsValue(0L)); assertFalse(equatables.containsValue(1L)); }
@Test public void testEntireSet() { EquatableValueSet equatables = EquatableValueSet.all(ID); assertEquals(equatables.getType(), ID); assertFalse(equatables.isNone()); assertTrue(equatables.isAll()); assertFalse(equatables.isSingleValue()); assertFalse(equatables.isWhiteList()); assertEquals(equatables.getValues().size(), 0); assertEquals(equatables.complement(), EquatableValueSet.none(ID)); assertTrue(equatables.containsValue(0L)); assertTrue(equatables.containsValue(1L)); }
@Test public void testEmptySet() { EquatableValueSet equatables = EquatableValueSet.none(ID); assertEquals(equatables.getType(), ID); assertTrue(equatables.isNone()); assertFalse(equatables.isAll()); assertFalse(equatables.isSingleValue()); assertTrue(equatables.isWhiteList()); assertEquals(equatables.getValues().size(), 0); assertEquals(equatables.complement(), EquatableValueSet.all(ID)); assertFalse(equatables.containsValue(0L)); assertFalse(equatables.containsValue(1L)); }
@Test public void testSingleValue() { EquatableValueSet equatables = EquatableValueSet.of(ID, 10L); EquatableValueSet complement = (EquatableValueSet) EquatableValueSet.all(ID).subtract(equatables); // Whitelist assertEquals(equatables.getType(), ID); assertFalse(equatables.isNone()); assertFalse(equatables.isAll()); assertTrue(equatables.isSingleValue()); assertTrue(equatables.isWhiteList()); assertTrue(Iterables.elementsEqual(equatables.getValues(), ImmutableList.of(10L))); assertEquals(equatables.complement(), complement); assertFalse(equatables.containsValue(0L)); assertFalse(equatables.containsValue(1L)); assertTrue(equatables.containsValue(10L)); // Blacklist assertEquals(complement.getType(), ID); assertFalse(complement.isNone()); assertFalse(complement.isAll()); assertFalse(complement.isSingleValue()); assertFalse(complement.isWhiteList()); assertTrue(Iterables.elementsEqual(complement.getValues(), ImmutableList.of(10L))); assertEquals(complement.complement(), equatables); assertTrue(complement.containsValue(0L)); assertTrue(complement.containsValue(1L)); assertFalse(complement.containsValue(10L)); }
@Test public void testSingleValue() { EquatableValueSet equatables = EquatableValueSet.of(ID, 10L); EquatableValueSet complement = (EquatableValueSet) EquatableValueSet.all(ID).subtract(equatables); // Whitelist assertEquals(equatables.getType(), ID); assertFalse(equatables.isNone()); assertFalse(equatables.isAll()); assertTrue(equatables.isSingleValue()); assertTrue(equatables.isWhiteList()); assertTrue(Iterables.elementsEqual(equatables.getValues(), ImmutableList.of(10L))); assertEquals(equatables.complement(), complement); assertFalse(equatables.containsValue(0L)); assertFalse(equatables.containsValue(1L)); assertTrue(equatables.containsValue(10L)); // Blacklist assertEquals(complement.getType(), ID); assertFalse(complement.isNone()); assertFalse(complement.isAll()); assertFalse(complement.isSingleValue()); assertFalse(complement.isWhiteList()); assertTrue(Iterables.elementsEqual(complement.getValues(), ImmutableList.of(10L))); assertEquals(complement.complement(), equatables); assertTrue(complement.containsValue(0L)); assertTrue(complement.containsValue(1L)); assertFalse(complement.containsValue(10L)); }