Optimized toArray method for ArrayDeque and ListBuilder #KT-42720

2021-01-22 13:12:50 +03:00
parent c2f6a2cb96
commit 421cb6971d
8 changed files with 194 additions and 23 deletions
@@ -0,0 +1,60 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package test.collections
+
+import kotlin.collections.builders.*
+import kotlin.test.Test
+import kotlin.test.assertSame
+import kotlin.test.assertTrue
+
+@Suppress("INVISIBLE_MEMBER")
+class ListBuilderTest {
+    @Test
+    @Suppress("PLATFORM_CLASS_MAPPED_TO_KOTLIN", "UNCHECKED_CAST")
+    fun toArray() {
+        val numberOfElements = 5
+        val expected = ArrayList<Int>().apply { addAll(0 until numberOfElements) }
+        val builder = ListBuilder<Int>().apply { addAll(0 until numberOfElements) }
+
+        fun testToArray(getter: List<Int>.() -> Array<Any?>) {
+            assertTrue(expected.getter() contentEquals builder.getter())
+            assertTrue(expected.subList(2, 4).getter() contentEquals builder.subList(2, 4).getter())
+        }
+
+        testToArray { (this as java.util.Collection<Int>).toArray() }
+    }
+
+    @Test
+    @Suppress("PLATFORM_CLASS_MAPPED_TO_KOTLIN", "UNCHECKED_CAST")
+    fun toArrayDestination() {
+        val numberOfElements = 5
+        val expected = ArrayList<Int>().apply { addAll(0 until numberOfElements) }
+        val builder = ListBuilder<Int>().apply { addAll(0 until numberOfElements) }
+
+        fun testToArray(destSize: Int, getter: List<Int>.(Array<Int>) -> Array<Int>) {
+            repeat(2) { index ->
+                val expectedDest = Array(destSize) { -it - 1 }
+                val builderDest = Array(destSize) { -it - 1 }
+
+                val takeSubList = index == 1
+                val expectedResult = (if (!takeSubList) expected else expected.subList(2, 4)).getter(expectedDest)
+                val builderResult = (if (!takeSubList) builder else builder.subList(2, 4)).getter(builderDest)
+
+                if (expectedResult.size <= expectedDest.size) {
+                    assertSame(expectedDest, expectedResult)
+                    assertSame(builderDest, builderResult)
+                }
+                assertTrue(expectedResult contentEquals builderResult)
+            }
+        }
+
+        testToArray(0) { (this as java.util.Collection<Int>).toArray(it) }
+        testToArray(numberOfElements - 1) { (this as java.util.Collection<Int>).toArray(it) }
+        testToArray(numberOfElements) { (this as java.util.Collection<Int>).toArray(it) }
+        testToArray(numberOfElements + 1) { (this as java.util.Collection<Int>).toArray(it) }
+        testToArray(numberOfElements + 2) { (this as java.util.Collection<Int>).toArray(it) }
+    }
+}