Update rocketpy/Function.py

Added try/except block to invert method

Co-authored-by: phmbressan <[email protected]>

Author: gautamsaiy

rocketpy/Function.py

@@ -2146,7 +2146,14 @@ def invert(self, lower=None, upper=None, datapoints=100):
         for i in range(datapoints):
             yData[i] = self.getValue(xData[i])
 
-        return Function(np.concatenate(([yData], [xData])).transpose(),
+        try:
+            return Function(
+                np.concatenate(([yData], [xData])).transpose(),
+                inputs=self.__outputs__,
+                outputs=self.__inputs__,
+            )
+        except ValueError:
+            raise ValueError("Check if given domain of the function is invertible.")
                     inputs=self.__outputs__,
                     outputs=self.__inputs__)