Add Root type

fintarin · Jul 18, 2023 · 3590682 · 3590682
1 parent fecc994
commit 3590682
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/include/fintamath/core/MathObjectTypes.hpp b/include/fintamath/core/MathObjectTypes.hpp
@@ -75,6 +75,7 @@ enum class MathObjectType : MathObjectTypeId {
   Lg,
   Exp,
   Sqrt,
+  Root,
   Sin,
   Cos,
   Tan,

diff --git a/include/fintamath/functions/powers/Root.hpp b/include/fintamath/functions/powers/Root.hpp
@@ -17,6 +17,10 @@ class Root : public IFunctionCRTP<INumber, Root, INumber, INumber> {
     return "root";
   }
 
+  static MathObjectTypeId getTypeIdStatic() {
+    return MathObjectTypeId(MathObjectType::Root);
+  }
+
 protected:
   std::unique_ptr<IMathObject> call(const ArgumentsRefVector &argsVect) const override;
 

diff --git a/tests/src/functions/powers/RootTests.cpp b/tests/src/functions/powers/RootTests.cpp
@@ -99,3 +99,8 @@ TEST(RootTests, callTest) {
   EXPECT_THROW(f(), InvalidInputFunctionException);
   EXPECT_THROW(f(Integer(1), Integer(1), Integer(1)), InvalidInputFunctionException);
 }
+
+TEST(RootTests, getTypeIdTest) {
+  EXPECT_EQ(Root::getTypeIdStatic(), MathObjectTypeId(MathObjectType::Root));
+  EXPECT_EQ(Root().getTypeId(), MathObjectTypeId(MathObjectType::Root));
+}