@@ -4,6 +4,7 @@ import io.kotest.assertions.throwables.shouldNotThrowAny
4
4
import io.kotest.matchers.nulls.shouldNotBeNull
5
5
import io.kotest.matchers.shouldBe
6
6
import kotlinx.datetime.LocalDateTime
7
+ import org.apache.commons.csv.CSVFormat
7
8
import org.jetbrains.kotlinx.dataframe.DataFrame
8
9
import org.jetbrains.kotlinx.dataframe.api.ParserOptions
9
10
import org.jetbrains.kotlinx.dataframe.api.allNulls
@@ -19,6 +20,7 @@ import org.jetbrains.kotlinx.dataframe.nrow
19
20
import org.jetbrains.kotlinx.dataframe.testCsv
20
21
import org.jetbrains.kotlinx.dataframe.testResource
21
22
import org.junit.Test
23
+ import java.io.File
22
24
import java.io.StringWriter
23
25
import java.util.Locale
24
26
import kotlin.reflect.KClass
@@ -74,8 +76,12 @@ class CsvTests {
74
76
}
75
77
76
78
@Test
77
- fun readCSVwithFrenchLocaleAndAlternativeDelimiter () {
78
- val df = DataFrame .readCSV(csvWithFrenchLocale, delimiter = ' ;' , parserOptions = ParserOptions (locale = Locale .FRENCH ))
79
+ fun readCsvWithFrenchLocaleAndAlternativeDelimiter () {
80
+ val df = DataFrame .readCSV(
81
+ url = csvWithFrenchLocale,
82
+ delimiter = ' ;' ,
83
+ parserOptions = ParserOptions (locale = Locale .FRENCH ),
84
+ )
79
85
80
86
df.ncol shouldBe 11
81
87
df.nrow shouldBe 5
@@ -141,7 +147,19 @@ class CsvTests {
141
147
@Test
142
148
fun `read first rows` () {
143
149
val expected =
144
- listOf (" untitled" , " user_id" , " name" , " duplicate" , " username" , " duplicate1" , " duplicate11" , " double" , " number" , " time" , " empty" )
150
+ listOf (
151
+ " untitled" ,
152
+ " user_id" ,
153
+ " name" ,
154
+ " duplicate" ,
155
+ " username" ,
156
+ " duplicate1" ,
157
+ " duplicate11" ,
158
+ " double" ,
159
+ " number" ,
160
+ " time" ,
161
+ " empty" ,
162
+ )
145
163
val dfHeader = DataFrame .readCSV(simpleCsv, readLines = 0 )
146
164
dfHeader.nrow shouldBe 0
147
165
dfHeader.columnNames() shouldBe expected
@@ -214,7 +232,7 @@ class CsvTests {
214
232
df.columnsCount() shouldBe 3
215
233
df.rowsCount() shouldBe 2
216
234
}
217
-
235
+
218
236
@Test
219
237
fun `write csv whitout header produce correct file` () {
220
238
val df = dataFrameOf(" a" , " b" , " c" )(
0 commit comments