Handling string NaN as Decimal(0)

alasco-tech · Jun 28, 2024 · 774a576 · 774a576
1 parent 81a79b5
commit 774a576
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pytests/test_money.py b/pytests/test_money.py
@@ -26,6 +26,12 @@ def test_init_float():
     assert one_million_dollars.amount == Decimal("1000000")
 
 
+def test_init_nan():
+    nan_money = Money("NaN")
+
+    assert nan_money.amount == Decimal(0)
+
+
 def test_repr():
     assert repr(Money(Decimal("1000000"))) == "Money('1000000')"
     assert repr(Money(Decimal("2.000"))) == "Money('2.000')"

diff --git a/src/decimals.rs b/src/decimals.rs
@@ -16,6 +16,11 @@ pub fn get_decimal(obj: Bound<PyAny>) -> PyResult<Decimal> {
         Ok(amount)
     } else if let Ok(f) = obj.extract::<f64>() {
         Ok(Decimal::from_f64(f).unwrap())
+    } else if let Ok(s) = obj.extract::<&str>() {
+        match s {
+            "NaN" => Ok(Decimal::new(0, 0)),
+            _ => Err(PyValueError::new_err("Invalid decimal")),
+        }
     } else {
         Err(PyValueError::new_err("Invalid decimal"))
     }