assertEstimations for LocalTime

Author: Kopilov

dataframe-arrow/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/arrow.kt

@@ -57,6 +57,7 @@ import java.nio.file.Files
 import java.time.Duration
 import java.time.LocalDate
 import java.time.LocalDateTime
+import java.time.LocalTime
 import kotlin.reflect.typeOf
 
 public class ArrowFeather : SupportedFormat {
@@ -161,10 +162,11 @@ private fun DateDayVector.values(range: IntRange): List<LocalDate?> = range.map
 }
 private fun DateMilliVector.values(range: IntRange): List<LocalDateTime?> = range.map { getObject(it) }
 
-private fun TimeNanoVector.values(range: IntRange): List<Long?> = range.map { getObject(it) }
-private fun TimeMicroVector.values(range: IntRange): List<Long?> = range.map { getObject(it) }
-private fun TimeMilliVector.values(range: IntRange): List<LocalDateTime?> = range.map { getObject(it) }
-private fun TimeSecVector.values(range: IntRange): List<Int?> = range.map { getObject(it) }
+private fun TimeNanoVector.values(range: IntRange): List<LocalTime?> = range.map { LocalTime.ofNanoOfDay(get(it)) }
+private fun TimeMicroVector.values(range: IntRange): List<LocalTime?> = range.map { LocalTime.ofNanoOfDay(get(it) * 1000) }
+private fun TimeMilliVector.values(range: IntRange): List<LocalTime?> = range.map { LocalTime.ofNanoOfDay(get(it).toLong() * 1000_000) }
+private fun TimeSecVector.values(range: IntRange): List<LocalTime?> = range.map { LocalTime.ofSecondOfDay(get(it).toLong()) }
+
 private fun StructVector.values(range: IntRange): List<Map<String, Any?>?> = range.map { getObject(it) }
 
 private fun VarCharVector.values(range: IntRange): List<String?> = range.map {

dataframe-arrow/src/test/kotlin/exampleEstimatesAssertions.kt

@@ -1,11 +1,11 @@
 import io.kotest.matchers.shouldBe
 import org.jetbrains.kotlinx.dataframe.AnyFrame
 import org.jetbrains.kotlinx.dataframe.DataColumn
-import org.jetbrains.kotlinx.dataframe.api.forEach
 import org.jetbrains.kotlinx.dataframe.api.forEachIndexed
 import java.math.BigInteger
 import java.time.LocalDate
 import java.time.LocalDateTime
+import java.time.LocalTime
 import java.time.ZoneOffset
 import kotlin.math.absoluteValue
 import kotlin.math.pow
@@ -118,9 +118,28 @@ internal fun assertEstimations(exampleFrame: AnyFrame) {
         element shouldBe LocalDateTime.ofEpochSecond(iBatch(i).toLong() * 60 * 60 * 24 * 30, 0, ZoneOffset.UTC)
     }
 
-    val timeSecCol = exampleFrame["time32_seconds"]
-    val timeMilliCol = exampleFrame["time32_milli"]
+    val timeSecCol = exampleFrame["time32_seconds"] as DataColumn<LocalTime?>
+    timeSecCol.type() shouldBe typeOf<LocalTime?>()
+    timeSecCol.forEachIndexed { i, element ->
+        element shouldBe LocalTime.ofSecondOfDay(iBatch(i).toLong())
+    }
+
+    val timeMilliCol = exampleFrame["time32_milli"] as DataColumn<LocalTime?>
+    timeMilliCol.type() shouldBe typeOf<LocalTime?>()
+    timeMilliCol.forEachIndexed { i, element ->
+        element shouldBe LocalTime.ofNanoOfDay(iBatch(i).toLong() * 1000_000)
+    }
+
+    val timeMicroCol = exampleFrame["time64_micro"] as DataColumn<LocalTime?>
+    timeMicroCol.type() shouldBe typeOf<LocalTime?>()
+    timeMicroCol.forEachIndexed { i, element ->
+        element shouldBe LocalTime.ofNanoOfDay(iBatch(i).toLong() * 1000)
+    }
+
+    val timeNanoCol = exampleFrame["time64_nano"] as DataColumn<LocalTime?>
+    timeNanoCol.type() shouldBe typeOf<LocalTime?>()
+    timeNanoCol.forEachIndexed { i, element ->
+        element shouldBe LocalTime.ofNanoOfDay(iBatch(i).toLong())
+    }
 
-    val timeMicroCol = exampleFrame["time64_micro"]
-    val timeNanoCol = exampleFrame["time64_nano"]
 }