typelevel#787 - tests have ordering and precision issues when run on …

…clusters - inifinity protection
chris-twiner · Apr 10, 2024 · 986891a · 986891a
1 parent e582962
commit 986891a
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/dataset/src/test/scala/frameless/functions/DoubleBehaviourUtils.scala b/dataset/src/test/scala/frameless/functions/DoubleBehaviourUtils.scala
@@ -21,13 +21,17 @@ object DoubleBehaviourUtils {
     case null => None
     case d: Double =>
       nanHandler(d).map { d =>
-        BigDecimal(d).setScale(
-          6,
-          if (d > 0)
-            BigDecimal.RoundingMode.FLOOR
-          else
-            BigDecimal.RoundingMode.CEILING
-        )
+        if (d == Double.NegativeInfinity || d == Double.PositiveInfinity)
+          BigDecimal("1000000.000000") * (if (d == Double.PositiveInfinity) 1
+                                          else -1)
+        else
+          BigDecimal(d).setScale(
+            6,
+            if (d > 0)
+              BigDecimal.RoundingMode.FLOOR
+            else
+              BigDecimal.RoundingMode.CEILING
+          )
       }
     case _ => ???
   }