Skip to content

Commit e981869

Browse files
committed
Move Arrow tests to package
1 parent c0bd6cf commit e981869

File tree

3 files changed

+9
-7
lines changed

3 files changed

+9
-7
lines changed

dataframe-arrow/src/test/kotlin/ArrowKtTest.kt renamed to dataframe-arrow/src/test/kotlin/org/jetbrains/kotlinx/dataframe/io/ArrowKtTest.kt

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
package org.jetbrains.kotlinx.dataframe.io
2+
13
import io.kotest.assertions.throwables.shouldThrow
24
import io.kotest.matchers.collections.shouldContain
35
import io.kotest.matchers.shouldBe
@@ -14,12 +16,6 @@ import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
1416
import org.jetbrains.kotlinx.dataframe.api.map
1517
import org.jetbrains.kotlinx.dataframe.api.remove
1618
import org.jetbrains.kotlinx.dataframe.api.toColumn
17-
import org.jetbrains.kotlinx.dataframe.io.ArrowWriter
18-
import org.jetbrains.kotlinx.dataframe.io.arrowWriter
19-
import org.jetbrains.kotlinx.dataframe.io.readArrowFeather
20-
import org.jetbrains.kotlinx.dataframe.io.readArrowIPC
21-
import org.jetbrains.kotlinx.dataframe.io.saveArrowIPCToByteArray
22-
import org.jetbrains.kotlinx.dataframe.io.writeArrowFeather
2319
import org.junit.Test
2420
import java.io.File
2521
import java.net.URL
@@ -135,7 +131,8 @@ internal class ArrowKtTest {
135131
citiesDeserialized["area"].type() shouldBe typeOf<Float>()
136132
citiesDeserialized["settled"].type() shouldBe typeOf<LocalDateTime>()
137133
shouldThrow<IllegalArgumentException> { citiesDeserialized["page_in_wiki"] }
138-
citiesDeserialized["film_in_youtube"] shouldBe DataColumn.createValueColumn("film_in_youtube", arrayOfNulls<String>(citiesExampleFrame.rowsCount()).asList())
134+
citiesDeserialized["film_in_youtube"] shouldBe DataColumn.createValueColumn("film_in_youtube", arrayOfNulls<String>(
135+
citiesExampleFrame.rowsCount()).asList())
139136
}
140137

141138
@Test

dataframe-arrow/src/test/kotlin/exampleEstimatesAssertions.kt renamed to dataframe-arrow/src/test/kotlin/org/jetbrains/kotlinx/dataframe/io/exampleEstimatesAssertions.kt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
package org.jetbrains.kotlinx.dataframe.io
2+
13
import io.kotest.matchers.shouldBe
24
import org.jetbrains.kotlinx.dataframe.AnyFrame
35
import org.jetbrains.kotlinx.dataframe.DataColumn
@@ -14,6 +16,7 @@ import kotlin.reflect.typeOf
1416

1517
/**
1618
* Assert that we have got the same data that was originally saved on example creation.
19+
* Example generation project is currently located at https://github.com/Kopilov/arrow_example
1720
*/
1821
internal fun assertEstimations(exampleFrame: AnyFrame, expectedNullable: Boolean, hasNulls: Boolean) {
1922
/**

dataframe-arrow/src/test/kotlin/examplesToWrite.kt renamed to dataframe-arrow/src/test/kotlin/org/jetbrains/kotlinx/dataframe/io/examplesToWrite.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
package org.jetbrains.kotlinx.dataframe.io
2+
13
import org.jetbrains.kotlinx.dataframe.DataColumn
24
import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
35
import java.net.URL

0 commit comments

Comments
 (0)