fix domain for log

Author: tswast

bigframes/core/compile/polars/operations/numeric_ops.py

@@ -46,7 +46,7 @@ def ln_op_impl(
 ) -> pl.Expr:
     import polars as pl
 
-    return pl.when(input < 0).then(float("nan")).otherwise(input.log())
+    return pl.when(input <= 0).then(float("nan")).otherwise(input.log())
 
 
 @polars_compiler.register_op(numeric_ops.Log10Op)
@@ -57,7 +57,7 @@ def log10_op_impl(
 ) -> pl.Expr:
     import polars as pl
 
-    return pl.when(input < 0).then(float("nan")).otherwise(input.log(base=10))
+    return pl.when(input <= 0).then(float("nan")).otherwise(input.log(base=10))
 
 
 @polars_compiler.register_op(numeric_ops.Log1pOp)
@@ -68,7 +68,7 @@ def log1p_op_impl(
 ) -> pl.Expr:
     import polars as pl
 
-    return pl.when(input < -1).then(float("nan")).otherwise((input + 1).log())
+    return pl.when(input <= -1).then(float("nan")).otherwise((input + 1).log())
 
 
 @polars_compiler.register_op(numeric_ops.SinOp)
@@ -88,4 +88,4 @@ def sqrt_op_impl(
 ) -> pl.Expr:
     import polars as pl
 
-    return pl.when(input < 0).then(float("nan")).otherwise(input.sqrt())
+    return pl.when(input <= 0).then(float("nan")).otherwise(input.sqrt())

bigframes/core/compile/sqlglot/expressions/numeric_ops.py

@@ -158,7 +158,7 @@ def _(expr: TypedExpr) -> sge.Expression:
     return sge.Case(
         ifs=[
             sge.If(
-                this=expr.expr < sge.convert(0),
+                this=expr.expr <= sge.convert(0),
                 true=constants._NAN,
             )
         ],
@@ -171,7 +171,7 @@ def _(expr: TypedExpr) -> sge.Expression:
     return sge.Case(
         ifs=[
             sge.If(
-                this=expr.expr < sge.convert(0),
+                this=expr.expr <= sge.convert(0),
                 true=constants._NAN,
             )
         ],
@@ -184,7 +184,7 @@ def _(expr: TypedExpr) -> sge.Expression:
     return sge.Case(
         ifs=[
             sge.If(
-                this=expr.expr < sge.convert(-1),
+                this=expr.expr <= sge.convert(-1),
                 true=constants._NAN,
             )
         ],
@@ -207,7 +207,7 @@ def _(expr: TypedExpr) -> sge.Expression:
     return sge.Case(
         ifs=[
             sge.If(
-                this=expr.expr < sge.convert(0),
+                this=expr.expr <= sge.convert(0),
                 true=constants._NAN,
             )
         ],