pylegend 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

pylegend/core/language/{operations → shared/operations}/number_operation_expressions.py

@@ -16,14 +16,16 @@ from pylegend._typing import (
     PyLegendSequence,
     PyLegendDict,
 )
-from pylegend.core.language.expression import (
+from pylegend.core.language.shared.expression import (
     PyLegendExpressionNumberReturn,
     PyLegendExpressionIntegerReturn,
     PyLegendExpressionBooleanReturn,
 )
-from pylegend.core.language.operations.binary_expression import PyLegendBinaryExpression
-from pylegend.core.language.operations.unary_expression import PyLegendUnaryExpression
+from pylegend.core.language.shared.operations.binary_expression import PyLegendBinaryExpression
+from pylegend.core.language.shared.operations.unary_expression import PyLegendUnaryExpression
+from pylegend.core.language.shared.helpers import generate_pure_functional_call
 from pylegend.core.tds.tds_frame import FrameToSqlConfig
+from pylegend.core.tds.tds_frame import FrameToPureConfig
 from pylegend.core.sql.metamodel import (
     Expression,
     QuerySpecification,
@@ -97,15 +99,26 @@ class PyLegendNumberAddExpression(PyLegendBinaryExpression, PyLegendExpressionNu
     ) -> Expression:
         return ArithmeticExpression(ArithmeticType.ADD, expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} + {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberAddExpression.__to_sql_func
+            PyLegendNumberAddExpression.__to_sql_func,
+            PyLegendNumberAddExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_both_operands(self, config)
+
 
 class PyLegendNumberMultiplyExpression(PyLegendBinaryExpression, PyLegendExpressionNumberReturn):
 
@@ -118,15 +131,26 @@ class PyLegendNumberMultiplyExpression(PyLegendBinaryExpression, PyLegendExpress
     ) -> Expression:
         return ArithmeticExpression(ArithmeticType.MULTIPLY, expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} * {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberMultiplyExpression.__to_sql_func
+            PyLegendNumberMultiplyExpression.__to_sql_func,
+            PyLegendNumberMultiplyExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_both_operands(self, config)
+
 
 class PyLegendNumberDivideExpression(PyLegendBinaryExpression, PyLegendExpressionNumberReturn):
 
@@ -139,15 +163,23 @@ class PyLegendNumberDivideExpression(PyLegendBinaryExpression, PyLegendExpressio
     ) -> Expression:
         return ArithmeticExpression(ArithmeticType.DIVIDE, expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"{op1_expr}->map(op | $op / {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberDivideExpression.__to_sql_func
+            PyLegendNumberDivideExpression.__to_sql_func,
+            PyLegendNumberDivideExpression.__to_pure_func
         )
 
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_second_operand(self, config)
+
 
 class PyLegendNumberSubtractExpression(PyLegendBinaryExpression, PyLegendExpressionNumberReturn):
 
@@ -160,15 +192,26 @@ class PyLegendNumberSubtractExpression(PyLegendBinaryExpression, PyLegendExpress
     ) -> Expression:
         return ArithmeticExpression(ArithmeticType.SUBTRACT, expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} - {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberSubtractExpression.__to_sql_func
+            PyLegendNumberSubtractExpression.__to_sql_func,
+            PyLegendNumberSubtractExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_both_operands(self, config)
+
 
 class PyLegendNumberLessThanExpression(PyLegendBinaryExpression, PyLegendExpressionBooleanReturn):
 
@@ -181,15 +224,23 @@ class PyLegendNumberLessThanExpression(PyLegendBinaryExpression, PyLegendExpress
     ) -> Expression:
         return ComparisonExpression(expression1, expression2, ComparisonOperator.LESS_THAN)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} < {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionBooleanReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberLessThanExpression.__to_sql_func
+            PyLegendNumberLessThanExpression.__to_sql_func,
+            PyLegendNumberLessThanExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
 
 class PyLegendNumberLessThanEqualExpression(PyLegendBinaryExpression, PyLegendExpressionBooleanReturn):
 
@@ -202,15 +253,23 @@ class PyLegendNumberLessThanEqualExpression(PyLegendBinaryExpression, PyLegendEx
     ) -> Expression:
         return ComparisonExpression(expression1, expression2, ComparisonOperator.LESS_THAN_OR_EQUAL)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} <= {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionBooleanReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberLessThanEqualExpression.__to_sql_func
+            PyLegendNumberLessThanEqualExpression.__to_sql_func,
+            PyLegendNumberLessThanEqualExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
 
 class PyLegendNumberGreaterThanExpression(PyLegendBinaryExpression, PyLegendExpressionBooleanReturn):
 
@@ -223,15 +282,23 @@ class PyLegendNumberGreaterThanExpression(PyLegendBinaryExpression, PyLegendExpr
     ) -> Expression:
         return ComparisonExpression(expression1, expression2, ComparisonOperator.GREATER_THAN)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} > {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionBooleanReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberGreaterThanExpression.__to_sql_func
+            PyLegendNumberGreaterThanExpression.__to_sql_func,
+            PyLegendNumberGreaterThanExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
 
 class PyLegendNumberGreaterThanEqualExpression(PyLegendBinaryExpression, PyLegendExpressionBooleanReturn):
 
@@ -244,15 +311,23 @@ class PyLegendNumberGreaterThanEqualExpression(PyLegendBinaryExpression, PyLegen
     ) -> Expression:
         return ComparisonExpression(expression1, expression2, ComparisonOperator.GREATER_THAN_OR_EQUAL)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return f"({op1_expr} >= {op2_expr})"
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionBooleanReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberGreaterThanEqualExpression.__to_sql_func
+            PyLegendNumberGreaterThanEqualExpression.__to_sql_func,
+            PyLegendNumberGreaterThanEqualExpression.__to_pure_func
         )
 
+    def is_non_nullable(self) -> bool:
+        return True
+
 
 class PyLegendNumberNegativeExpression(PyLegendUnaryExpression, PyLegendExpressionNumberReturn):
 
@@ -264,12 +339,17 @@ class PyLegendNumberNegativeExpression(PyLegendUnaryExpression, PyLegendExpressi
     ) -> Expression:
         return NegativeExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("minus", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberNegativeExpression.__to_sql_func
+            PyLegendNumberNegativeExpression.__to_sql_func,
+            PyLegendNumberNegativeExpression.__to_pure_func
         )
 
 
@@ -283,12 +363,17 @@ class PyLegendNumberAbsoluteExpression(PyLegendUnaryExpression, PyLegendExpressi
     ) -> Expression:
         return AbsoluteExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("abs", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberAbsoluteExpression.__to_sql_func
+            PyLegendNumberAbsoluteExpression.__to_sql_func,
+            PyLegendNumberAbsoluteExpression.__to_pure_func
         )
 
 
@@ -303,15 +388,23 @@ class PyLegendNumberPowerExpression(PyLegendBinaryExpression, PyLegendExpression
     ) -> Expression:
         return PowerExpression(expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("pow", [op1_expr, op2_expr], auto_map=True)
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberPowerExpression.__to_sql_func
+            PyLegendNumberPowerExpression.__to_sql_func,
+            PyLegendNumberPowerExpression.__to_pure_func
         )
 
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_second_operand(self, config)
+
 
 class PyLegendNumberCeilExpression(PyLegendUnaryExpression, PyLegendExpressionIntegerReturn):
 
@@ -323,12 +416,17 @@ class PyLegendNumberCeilExpression(PyLegendUnaryExpression, PyLegendExpressionIn
     ) -> Expression:
         return CeilExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("ceiling", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionIntegerReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberCeilExpression.__to_sql_func
+            PyLegendNumberCeilExpression.__to_sql_func,
+            PyLegendNumberCeilExpression.__to_pure_func
         )
 
 
@@ -342,12 +440,17 @@ class PyLegendNumberFloorExpression(PyLegendUnaryExpression, PyLegendExpressionI
     ) -> Expression:
         return FloorExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("floor", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionIntegerReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberFloorExpression.__to_sql_func
+            PyLegendNumberFloorExpression.__to_sql_func,
+            PyLegendNumberFloorExpression.__to_pure_func
         )
 
 
@@ -361,12 +464,17 @@ class PyLegendNumberSqrtExpression(PyLegendUnaryExpression, PyLegendExpressionNu
     ) -> Expression:
         return SqrtExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("sqrt", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberSqrtExpression.__to_sql_func
+            PyLegendNumberSqrtExpression.__to_sql_func,
+            PyLegendNumberSqrtExpression.__to_pure_func
         )
 
 
@@ -380,12 +488,17 @@ class PyLegendNumberCbrtExpression(PyLegendUnaryExpression, PyLegendExpressionNu
     ) -> Expression:
         return CbrtExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("cbrt", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberCbrtExpression.__to_sql_func
+            PyLegendNumberCbrtExpression.__to_sql_func,
+            PyLegendNumberCbrtExpression.__to_pure_func
         )
 
 
@@ -399,12 +512,17 @@ class PyLegendNumberExpExpression(PyLegendUnaryExpression, PyLegendExpressionNum
     ) -> Expression:
         return ExpExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("exp", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberExpExpression.__to_sql_func
+            PyLegendNumberExpExpression.__to_sql_func,
+            PyLegendNumberExpExpression.__to_pure_func
         )
 
 
@@ -418,12 +536,17 @@ class PyLegendNumberLogExpression(PyLegendUnaryExpression, PyLegendExpressionNum
     ) -> Expression:
         return LogExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("log", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberLogExpression.__to_sql_func
+            PyLegendNumberLogExpression.__to_sql_func,
+            PyLegendNumberLogExpression.__to_pure_func
         )
 
 
@@ -438,15 +561,23 @@ class PyLegendNumberRemainderExpression(PyLegendBinaryExpression, PyLegendExpres
     ) -> Expression:
         return RemainderExpression(expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("rem", [op1_expr, op2_expr], auto_map=True)
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberRemainderExpression.__to_sql_func
+            PyLegendNumberRemainderExpression.__to_sql_func,
+            PyLegendNumberRemainderExpression.__to_pure_func
         )
 
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_second_operand(self, config)
+
 
 class PyLegendNumberRoundExpression(PyLegendBinaryExpression, PyLegendExpressionNumberReturn):
 
@@ -459,13 +590,20 @@ class PyLegendNumberRoundExpression(PyLegendBinaryExpression, PyLegendExpression
     ) -> Expression:
         return RoundExpression(expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        if op2_expr == "0":
+            return generate_pure_functional_call("round", [op1_expr], auto_map=True)
+        return generate_pure_functional_call("round", [f"cast({op1_expr}, @Float)", op2_expr], auto_map=True)
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionIntegerReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberRoundExpression.__to_sql_func
+            PyLegendNumberRoundExpression.__to_sql_func,
+            PyLegendNumberRoundExpression.__to_pure_func
         )
 
 
@@ -479,12 +617,17 @@ class PyLegendNumberSineExpression(PyLegendUnaryExpression, PyLegendExpressionNu
     ) -> Expression:
         return SineExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("sin", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberSineExpression.__to_sql_func
+            PyLegendNumberSineExpression.__to_sql_func,
+            PyLegendNumberSineExpression.__to_pure_func
         )
 
 
@@ -498,12 +641,17 @@ class PyLegendNumberArcSineExpression(PyLegendUnaryExpression, PyLegendExpressio
     ) -> Expression:
         return ArcSineExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("asin", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberArcSineExpression.__to_sql_func
+            PyLegendNumberArcSineExpression.__to_sql_func,
+            PyLegendNumberArcSineExpression.__to_pure_func
         )
 
 
@@ -517,12 +665,17 @@ class PyLegendNumberCosineExpression(PyLegendUnaryExpression, PyLegendExpression
     ) -> Expression:
         return CosineExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("cos", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberCosineExpression.__to_sql_func
+            PyLegendNumberCosineExpression.__to_sql_func,
+            PyLegendNumberCosineExpression.__to_pure_func
         )
 
 
@@ -536,12 +689,17 @@ class PyLegendNumberArcCosineExpression(PyLegendUnaryExpression, PyLegendExpress
     ) -> Expression:
         return ArcCosineExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("acos", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberArcCosineExpression.__to_sql_func
+            PyLegendNumberArcCosineExpression.__to_sql_func,
+            PyLegendNumberArcCosineExpression.__to_pure_func
         )
 
 
@@ -555,12 +713,17 @@ class PyLegendNumberTanExpression(PyLegendUnaryExpression, PyLegendExpressionNum
     ) -> Expression:
         return TanExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("tan", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberTanExpression.__to_sql_func
+            PyLegendNumberTanExpression.__to_sql_func,
+            PyLegendNumberTanExpression.__to_pure_func
         )
 
 
@@ -574,12 +737,17 @@ class PyLegendNumberArcTanExpression(PyLegendUnaryExpression, PyLegendExpression
     ) -> Expression:
         return ArcTanExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("atan", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberArcTanExpression.__to_sql_func
+            PyLegendNumberArcTanExpression.__to_sql_func,
+            PyLegendNumberArcTanExpression.__to_pure_func
         )
 
 
@@ -594,15 +762,23 @@ class PyLegendNumberArcTan2Expression(PyLegendBinaryExpression, PyLegendExpressi
     ) -> Expression:
         return ArcTan2Expression(expression1, expression2)
 
+    @staticmethod
+    def __to_pure_func(op1_expr: str, op2_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("atan2", [op1_expr, op2_expr], auto_map=True)
+
     def __init__(self, operand1: PyLegendExpressionNumberReturn, operand2: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendBinaryExpression.__init__(
             self,
             operand1,
             operand2,
-            PyLegendNumberArcTan2Expression.__to_sql_func
+            PyLegendNumberArcTan2Expression.__to_sql_func,
+            PyLegendNumberArcTan2Expression.__to_pure_func
         )
 
+    def to_pure_expression(self, config: FrameToPureConfig) -> str:
+        return PyLegendBinaryExpression.to_pure_expression_with_to_one_on_second_operand(self, config)
+
 
 class PyLegendNumberCotExpression(PyLegendUnaryExpression, PyLegendExpressionNumberReturn):
 
@@ -614,10 +790,15 @@ class PyLegendNumberCotExpression(PyLegendUnaryExpression, PyLegendExpressionNum
     ) -> Expression:
         return CotExpression(expression)
 
+    @staticmethod
+    def __to_pure_func(op_expr: str, config: FrameToPureConfig) -> str:
+        return generate_pure_functional_call("cot", [op_expr], auto_map=True)
+
     def __init__(self, operand: PyLegendExpressionNumberReturn) -> None:
         PyLegendExpressionNumberReturn.__init__(self)
         PyLegendUnaryExpression.__init__(
             self,
             operand,
-            PyLegendNumberCotExpression.__to_sql_func
+            PyLegendNumberCotExpression.__to_sql_func,
+            PyLegendNumberCotExpression.__to_pure_func
         )