add tests

attack68 · Feb 10, 2024 · f60b2c1 · f60b2c1
1 parent 740f46f
commit f60b2c1
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 2 deletions.
diff --git a/src/dual/dual1.rs b/src/dual/dual1.rs
@@ -113,6 +113,34 @@ impl Dual {
         }
     }
 
+    fn __lt__(&self, other:DualOrF64) -> PyResult<bool> {
+        match other {
+            DualOrF64::Dual(d) => Ok(self < &d),
+            DualOrF64::F64(f) => Ok(self < &f)
+        }
+    }
+
+    fn __le__(&self, other:DualOrF64) -> PyResult<bool> {
+        match other {
+            DualOrF64::Dual(d) => Ok(self <= &d),
+            DualOrF64::F64(f) => Ok(self <= &f)
+        }
+    }
+
+    fn __gt__(&self, other:DualOrF64) -> PyResult<bool> {
+        match other {
+            DualOrF64::Dual(d) => Ok(self > &d),
+            DualOrF64::F64(f) => Ok(self > &f)
+        }
+    }
+
+    fn __ge__(&self, other:DualOrF64) -> PyResult<bool> {
+        match other {
+            DualOrF64::Dual(d) => Ok(self >= &d),
+            DualOrF64::F64(f) => Ok(self >= &f)
+        }
+    }
+
     fn __neg__(&self) -> Self {-self}
 
     fn __add__(&self, other: DualOrF64) -> Self {

diff --git a/tests/test_dualrs.py b/tests/test_dualrs.py
@@ -71,5 +71,20 @@ def test_eq_ne(x_1):
 
 def test_lt():
     assert Dual(1, ["x"], []) < Dual(2, ["y"], [])
-    assert Dual(1, "x") < 10
-    assert not Dual(1, "x") < 0
+    assert Dual(1, ["x"], []) < 10
+    assert 0.5 < Dual(1, ["x"], [])
+
+def test_le():
+    assert Dual(1.0, ["x"], []) <= Dual(1.0, ["y"], [])
+    assert Dual(1, ["x"], []) <= 1.0
+    assert 1.0 <= Dual(1.0, ["x"], [])
+
+def test_gt():
+    assert Dual(3, ["x"], []) > Dual(2, ["y"], [])
+    assert Dual(1, ["x"], []) > 0.5
+    assert 0.5 > Dual(0.3, ["x"], [])
+
+def test_ge():
+    assert Dual(1.0, ["x"], []) >= Dual(1.0, ["y"], [])
+    assert Dual(1, ["x"], []) >= 1.0
+    assert 1.0 >= Dual(1.0, ["x"], [])