14c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin/* 24c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved. 34c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 44c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 54c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * This code is free software; you can redistribute it and/or modify it 64c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * under the terms of the GNU General Public License version 2 only, as 74c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * published by the Free Software Foundation. 84c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 94c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * This code is distributed in the hope that it will be useful, but WITHOUT 104c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 114c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 124c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * version 2 for more details (a copy is included in the LICENSE file that 134c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * accompanied this code). 144c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 154c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * You should have received a copy of the GNU General Public License version 164c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 2 along with this work; if not, write to the Free Software Foundation, 174c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 184c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 194c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 204c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * or visit www.oracle.com if you need additional information or have any 214c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * questions. 224c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin */ 234c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinpackage java.util.stream; 244c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 254c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.ArrayList; 264c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.Arrays; 274c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.List; 284c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.PrimitiveIterator; 294c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.Spliterators; 304c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.function.Function; 314c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 324c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport org.testng.annotations.DataProvider; 334c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport org.testng.annotations.Test; 344c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 354c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin@Test 364c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinpublic class IntNodeTest extends OpTestCase { 374c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 384c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @DataProvider(name = "nodes") 394c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public Object[][] createSizes() { 404c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Object[]> params = new ArrayList<>(); 414c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 424c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int size : Arrays.asList(0, 1, 4, 15, 16, 17, 127, 128, 129, 1000)) { 434c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int[] array = new int[size]; 444c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int i = 0; i < array.length; i++) { 454c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin array[i] = i; 464c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 474c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 484c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Node<Integer>> nodes = new ArrayList<>(); 494c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 504c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(Nodes.node(array)); 514c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(degenerateTree(Spliterators.iterator(Arrays.spliterator(array)))); 524c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(tree(toList(array), l -> Nodes.node(toIntArray(l)))); 534c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(fill(array, Nodes.intBuilder(array.length))); 544c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(fill(array, Nodes.intBuilder())); 554c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 564c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (Node<Integer> node : nodes) { 574c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin params.add(new Object[]{array, node}); 584c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 594c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 604c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 614c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 624c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return params.toArray(new Object[0][]); 634c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 644c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 654c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin private static void assertEqualsListIntArray(List<Integer> list, int[] array) { 664c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(list.size(), array.length); 674c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int i = 0; i < array.length; i++) 684c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(array[i], (int) list.get(i)); 694c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 704c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 714c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin private List<Integer> toList(int[] a) { 724c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Integer> l = new ArrayList<>(); 734c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int i : a) { 744c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin l.add(i); 754c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 764c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 774c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return l; 784c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 794c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 804c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin private int[] toIntArray(List<Integer> l) { 814c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int[] a = new int[l.size()]; 824c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 834c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int i = 0; 844c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (Integer e : l) { 854c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin a[i++] = e; 864c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 874c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return a; 884c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 894c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 904c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin private Node.OfInt fill(int[] array, Node.Builder.OfInt nb) { 914c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nb.begin(array.length); 924c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int i : array) 934c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nb.accept(i); 944c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nb.end(); 954c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return nb.build(); 964c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 974c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 984c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin private Node.OfInt degenerateTree(PrimitiveIterator.OfInt it) { 994c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (!it.hasNext()) { 1004c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return Nodes.node(new int[0]); 1014c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1024c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1034c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int i = it.nextInt(); 1044c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (it.hasNext()) { 1054c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return new Nodes.ConcNode.OfInt(Nodes.node(new int[] {i}), degenerateTree(it)); 1064c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1074c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin else { 1084c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return Nodes.node(new int[] {i}); 1094c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1104c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1114c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1124c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin private Node.OfInt tree(List<Integer> l, Function<List<Integer>, Node.OfInt> m) { 1134c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (l.size() < 3) { 1144c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return m.apply(l); 1154c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1164c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin else { 1174c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return new Nodes.ConcNode.OfInt( 1184c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin tree(l.subList(0, l.size() / 2), m), 1194c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin tree(l.subList(l.size() / 2, l.size()), m)); 1204c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1214c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1224c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1234c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1244c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testAsArray(int[] array, Node.OfInt n) { 1254c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(n.asPrimitiveArray(), array); 1264c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1274c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1284c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1294c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testFlattenAsArray(int[] array, Node.OfInt n) { 1304c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(Nodes.flattenInt(n).asPrimitiveArray(), array); 1314c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1324c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1334c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1344c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testCopyTo(int[] array, Node.OfInt n) { 1354c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int[] copy = new int[(int) n.count()]; 1364c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin n.copyInto(copy, 0); 1374c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1384c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(copy, array); 1394c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1404c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1414c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes", groups = { "serialization-hostile" }) 1424c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testForEach(int[] array, Node.OfInt n) { 1434c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Integer> l = new ArrayList<>((int) n.count()); 1444c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin n.forEach((int e) -> { 1454c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin l.add(e); 1464c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin }); 1474c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1484c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEqualsListIntArray(l, array); 1494c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1504c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1514c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1524c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testStreams(int[] array, Node.OfInt n) { 1534c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin TestData.OfInt data = TestData.Factory.ofNode("Node", n); 1544c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1554c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin exerciseOps(data, s -> s); 1564c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin exerciseTerminalOps(data, s -> s.toArray()); 1574c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1584c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1594c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1604c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testSpliterator(int[] array, Node.OfInt n) { 1614c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin SpliteratorTestHelper.testIntSpliterator(n::spliterator); 1624c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1634c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1644c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1654c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testTruncate(int[] array, Node.OfInt n) { 1664c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int[] nums = new int[] { 0, 1, array.length / 2, array.length - 1, array.length }; 1674c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int start : nums) 1684c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int end : nums) { 1694c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (start < 0 || end < 0 || end < start || end > array.length) 1704c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin continue; 1714c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Node.OfInt slice = n.truncate(start, end, Integer[]::new); 1724c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int[] asArray = slice.asPrimitiveArray(); 1734c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int k = start; k < end; k++) 1744c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(array[k], asArray[k - start]); 1754c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1764c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1774c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin} 178