From cd0b6ec8f4c53ade3cc37fa654abb1a3f2a24801 Mon Sep 17 00:00:00 2001
From: Jolan Rensen <jolan.rensen@jetbrains.com>
Date: Tue, 19 Mar 2024 17:40:53 +0100
Subject: [PATCH] added toDataFrame for float- and double iterables

---
 .../org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt | 10 ++++++++++
 .../org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt |  6 ++++++
 2 files changed, 16 insertions(+)

diff --git a/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt b/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt
index 4cf7f23198..ec3832563b 100644
--- a/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt
+++ b/core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt
@@ -232,6 +232,16 @@ public inline fun <reified B : Boolean?> Iterable<B>.toDataFrame(): DataFrame<Va
     ValueProperty<B>::value from { it }
 }.cast()
 
+@JvmName("toDataFrameFloat")
+public inline fun <reified F : Float?> Iterable<F>.toDataFrame(): DataFrame<ValueProperty<F>> = toDataFrame {
+    ValueProperty<F>::value from { it }
+}.cast()
+
+@JvmName("toDataFrameDouble")
+public inline fun <reified D : Double?> Iterable<D>.toDataFrame(): DataFrame<ValueProperty<D>> = toDataFrame {
+    ValueProperty<D>::value from { it }
+}.cast()
+
 @JvmName("toDataFrameUByte")
 public inline fun <reified U : UByte?> Iterable<U>.toDataFrame(): DataFrame<ValueProperty<U>> = toDataFrame {
     ValueProperty<U>::value from { it }
diff --git a/core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt b/core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt
index 0b42c3dc47..f5247ccc64 100644
--- a/core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt
+++ b/core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt
@@ -223,6 +223,12 @@ class CreateDataFrameTests {
 
         val int = listOf(1, 2, 3)
         int.toDataFrame().alsoDebug() shouldBe dataFrameOf("value")(*int.toTypedArray())
+
+        val doubles = listOf(1.0, 2.0, 3.0)
+        doubles.toDataFrame().alsoDebug() shouldBe dataFrameOf("value")(*doubles.toTypedArray())
+
+        val floats = listOf(1.0f, 2.0f, 3.0f)
+        floats.toDataFrame().alsoDebug() shouldBe dataFrameOf("value")(*floats.toTypedArray())
     }
 
     @Ignore