1a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin/* 2a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved. 3a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * 5a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * This code is free software; you can redistribute it and/or modify it 6a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * under the terms of the GNU General Public License version 2 only, as 7a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * published by the Free Software Foundation. 8a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * 9a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * This code is distributed in the hope that it will be useful, but WITHOUT 10a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * version 2 for more details (a copy is included in the LICENSE file that 13a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * accompanied this code). 14a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * 15a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * You should have received a copy of the GNU General Public License version 16a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * 2 along with this work; if not, write to the Free Software Foundation, 17a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * 19a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * or visit www.oracle.com if you need additional information or have any 21a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin * questions. 22a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin */ 23a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinpackage org.openjdk.tests.java.util.stream; 24a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 25a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport org.testng.Assert; 26a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport org.testng.annotations.Test; 27a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 28a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.Arrays; 29a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.List; 30a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.Random; 31a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.concurrent.atomic.AtomicInteger; 32a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.function.IntConsumer; 33a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.stream.Collectors; 34a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport java.util.stream.IntStream; 35a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 36a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinimport static org.testng.Assert.assertEquals; 37a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 38a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin@Test 39a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkinpublic class IntPrimitiveOpsTests { 40a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 41a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testSum() { 42a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin long sum = IntStream.range(1, 10).filter(i -> i % 2 == 0).sum(); 43a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(sum, 20); 44a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 45a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 46a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testMap() { 47a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin long sum = IntStream.range(1, 10).filter(i -> i % 2 == 0).map(i -> i * 2).sum(); 48a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(sum, 40); 49a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 50a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 51a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testParSum() { 52a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin long sum = IntStream.range(1, 10).parallel().filter(i -> i % 2 == 0).sum(); 53a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(sum, 20); 54a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 55a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 56a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin @Test(groups = { "serialization-hostile" }) 57a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testTee() { 58a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] teeSum = new int[1]; 59a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin long sum = IntStream.range(1, 10).filter(i -> i % 2 == 0).peek(i -> { teeSum[0] = teeSum[0] + i; }).sum(); 60a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(teeSum[0], sum); 61a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 62a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 63a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin @Test(groups = { "serialization-hostile" }) 64a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testForEach() { 65a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] sum = new int[1]; 66a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin IntStream.range(1, 10).filter(i -> i % 2 == 0).forEach(i -> { sum[0] = sum[0] + i; }); 67a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(sum[0], 20); 68a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 69a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 70a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin @Test(groups = { "serialization-hostile" }) 71a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testParForEach() { 72a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin AtomicInteger ai = new AtomicInteger(0); 73a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin IntStream.range(1, 10).parallel().filter(i -> i % 2 == 0).forEach(ai::addAndGet); 74a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(ai.get(), 20); 75a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 76a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 77a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testBox() { 78a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin List<Integer> l = IntStream.range(1, 10).parallel().boxed().collect(Collectors.toList()); 79a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int sum = l.stream().reduce(0, (a, b) -> a + b); 80a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(sum, 45); 81a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 82a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 83a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testUnBox() { 84a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin long sum = Arrays.asList(1, 2, 3, 4, 5).stream().mapToInt(i -> (int) i).sum(); 85a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(sum, 15); 86a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 87a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 88a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testToArray() { 89a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 90a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] array = IntStream.range(1, 10).map(i -> i * 2).toArray(); 91a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array, new int[]{2, 4, 6, 8, 10, 12, 14, 16, 18}); 92a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 93a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 94a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 95a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] array = IntStream.range(1, 10).parallel().map(i -> i * 2).toArray(); 96a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array, new int[]{2, 4, 6, 8, 10, 12, 14, 16, 18}); 97a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 98a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 99a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 100a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testSort() { 101a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin Random r = new Random(); 102a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 103a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] content = IntStream.generate(() -> r.nextInt(100)).limit(10).toArray(); 104a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] sortedContent = content.clone(); 105a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin Arrays.sort(sortedContent); 106a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 107a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 108a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] array = Arrays.stream(content).sorted().toArray(); 109a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array, sortedContent); 110a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 111a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 112a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 113a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] array = Arrays.stream(content).parallel().sorted().toArray(); 114a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array, sortedContent); 115a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 116a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 117a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 118a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testSortSort() { 119a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin Random r = new Random(); 120a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 121a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] content = IntStream.generate(() -> r.nextInt(100)).limit(10).toArray(); 122a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] sortedContent = content.clone(); 123a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin Arrays.sort(sortedContent); 124a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 125a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 126a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] array = Arrays.stream(content).sorted().sorted().toArray(); 127a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array, sortedContent); 128a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 129a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 130a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 131a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] array = Arrays.stream(content).parallel().sorted().sorted().toArray(); 132a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array, sortedContent); 133a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 134a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 135a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 136a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testSequential() { 137a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 138a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] expected = IntStream.range(1, 1000).toArray(); 139a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 140a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin class AssertingConsumer implements IntConsumer { 141a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin private final int[] array; 142a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int offset; 143a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 144a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin AssertingConsumer(int[] array) { 145a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin this.array = array; 146a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 147a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 148a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin @Override 149a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void accept(int value) { 150a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(array[offset++], value); 151a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 152a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 153a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public int getCount() { return offset; } 154a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 155a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 156a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 157a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin AssertingConsumer consumer = new AssertingConsumer(expected); 158a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin IntStream.range(1, 1000).sequential().forEach(consumer); 159a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(expected.length, consumer.getCount()); 160a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 161a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 162a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 163a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin AssertingConsumer consumer = new AssertingConsumer(expected); 164a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin IntStream.range(1, 1000).parallel().sequential().forEach(consumer); 165a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin assertEquals(expected.length, consumer.getCount()); 166a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 167a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 168a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 169a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin public void testLimit() { 170a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] expected = IntStream.range(1, 10).toArray(); 171a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 172a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 173a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] actual = IntStream.iterate(1, i -> i + 1).limit(9).toArray(); 174a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin Assert.assertTrue(Arrays.equals(expected, actual)); 175a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 176a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 177a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin { 178a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin int[] actual = IntStream.range(1, 100).parallel().limit(9).toArray(); 179a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin Assert.assertTrue(Arrays.equals(expected, actual)); 180a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 181a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin } 182a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin 183a405b7ea164e472bcf6b75d9890021bd82e4d60dIgor Murashkin} 184