Support the comparison operators for Snowpark (finos#59)

Support the comparison operators for snowpark ===, =!=, <, >, <=, >=
Snowflake-Labs · Dec 13, 2023 · 28877eb · 28877eb
1 parent fa015ed
commit 28877eb
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/src/Morphir/Snowpark/MapFunctionsMapping.elm b/src/Morphir/Snowpark/MapFunctionsMapping.elm
@@ -60,6 +60,33 @@ mapFunctionsMapping value mapValue ctx =
                 operatorname = mapOperator optname
             in
             Scala.BinOp leftValue operatorname rightValue
+        ValueIR.Apply 
+            (TypeIR.Reference () ([["morphir"],["s","d","k"]],[["basics"]],["bool"]) [])
+            (ValueIR.Apply (TypeIR.Function 
+                        ()
+                        (TypeIR.Variable () ["t","0"])
+                        (TypeIR.Reference () ([["morphir"],["s","d","k"]],[["basics"]],["bool"]) [])
+                    )
+                    (ValueIR.Reference (TypeIR.Function 
+                                            ()
+                                            (TypeIR.Variable () ["t","0"])
+                                            (TypeIR.Function 
+                                                ()
+                                                (TypeIR.Variable () ["t","0"])
+                                                (TypeIR.Reference () ([["morphir"],["s","d","k"]],[["basics"]],["bool"]) [])
+                                            )
+                                        )
+                                ([["morphir"],["s","d","k"]],[["basics"]], optname)
+                    )
+                    left
+            )
+            right ->
+            let
+                leftValue = mapValue left ctx
+                rightValue = mapValue right ctx
+                operatorname = mapOperator optname
+            in
+            Scala.BinOp leftValue operatorname rightValue
         _ ->
             Scala.Literal (Scala.StringLit "To Do")
 

diff --git a/src/Morphir/Snowpark/Operatorsmaps.elm b/src/Morphir/Snowpark/Operatorsmaps.elm
@@ -15,5 +15,17 @@ mapOperator name =
             "/"
         ["float", "divide"] ->
             "/"
+        ["equal"] ->
+            "==="
+        ["not", "equal"] ->
+            "=!="
+        ["greater", "than"] ->
+            ">"
+        ["less", "than"] ->
+            "<"
+        ["less", "than", "or", "equal"] ->
+            "<="
+        ["greater", "than", "or", "equal"] ->
+            ">="
         _ ->
             "Unsupported"