Test and documentation related to signum function.

Signed-off-by: Lukasz Soszynski <[email protected]>
opensearch-project · Sep 30, 2024 · be667b4 · be667b4
1 parent b67a40f
commit be667b4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/ppl-spark-integration/README.md b/ppl-spark-integration/README.md
@@ -333,6 +333,7 @@ Limitation: Overriding existing field is unsupported, following queries throw ex
  - `source = table | eval a = 10 | fields a,b,c`
  - `source = table | eval a = a * 2 | stats avg(a)`
  - `source = table | eval a = abs(a) | where a > 0`
+ - `source = table | eval a = signum(a) | where a < 0`
 
 **Aggregations**
  - `source = table | stats avg(a) `

diff --git a/...scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanMathFunctionsTranslatorTestSuite.scala b/...scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanMathFunctionsTranslatorTestSuite.scala
@@ -191,4 +191,18 @@ class PPLLogicalPlanMathFunctionsTranslatorTestSuite
     val expectedPlan = Project(projectList, evalProject)
     comparePlans(expectedPlan, logPlan, false)
   }
+
+  test("test signum") {
+    val context = new CatalystPlanContext
+    val logPlan = planTransformer.visit(plan(pplParser, "source=t a = signum(b)", false), context)
+
+    val table = UnresolvedRelation(Seq("t"))
+    val filterExpr = EqualTo(
+      UnresolvedAttribute("a"),
+      UnresolvedFunction("signum", seq(UnresolvedAttribute("b")), isDistinct = false))
+    val filterPlan = Filter(filterExpr, table)
+    val projectList = Seq(UnresolvedStar(None))
+    val expectedPlan = Project(projectList, filterPlan)
+    comparePlans(expectedPlan, logPlan, false)
+  }
 }