@Override protected IntColumn.Builder builder() { return IntColumn.builder(ID); }
public static IntColumn ofAll(IntColumnId id, int... values) { return builder(id).addAll(values).build(); }
public static IntColumn of(IntColumnId id, int value) { return builder(id).add(value).build(); }
public static IntColumn ofAll(IntColumnId id, IntStream values) { return builder(id).addAll(values).build(); }
public static IntColumn ofAll(IntColumnId id, IntStream values) { return builder(id).addAll(values).build(); }
public static IntColumn of(IntColumnId id, int value) { return builder(id).add(value).build(); }
public static IntColumn ofAll(IntColumnId id, int... values) { return builder(id).addAll(values).build(); }
@Test(expected = IllegalArgumentException.class) public void failOnDifferingColumnSizes() { StringColumn oneRowColumn = StringColumn.builder(NAME).add("foo").build(); IntColumn threeRowColumn = IntColumn.builder(AGE).addAll(1, 2, 3).build(); DataFrame.ofAll(oneRowColumn, threeRowColumn); }
@Test public void ofAll() { StringColumn stringColumn = StringColumn.builder(NAME).build(); IntColumn intColumn = IntColumn.builder(AGE).build(); // enum variant assertEquals(2, DataFrame.ofAll(stringColumn, intColumn).getColumnCount()); // iterable variant Iterable<Column<?>> columns = Arrays.asList(stringColumn, intColumn); assertEquals(2, DataFrame.ofAll(columns).getColumnCount()); }
@Test public void withMetaData() { StringColumn stringColumn = StringColumn.builder(NAME).build(); IntColumn intColumn = IntColumn.builder(AGE).build(); DataFrame df = DataFrame.ofAll(stringColumn, intColumn); Map<String, String> metaData = LinkedHashMap.of("foo", "bar"); DataFrame withMetaData = df.withMetaData(metaData); assertEquals(metaData, withMetaData.getMetaData()); assertEquals(2, withMetaData.getColumnCount()); }
@Test public void iterator() { StringColumn stringColumn = StringColumn.builder(NAME).build(); IntColumn intColumn = IntColumn.builder(AGE).build(); Iterator<Column<?>> iterator = DataFrame.ofAll(stringColumn, intColumn).iterator(); assertTrue(iterator.hasNext()); iterator.next(); assertTrue(iterator.hasNext()); iterator.next(); assertFalse(iterator.hasNext()); }