diff --git a/src/libstd/cmp.rs b/src/libstd/cmp.rs index 110559ddcef..991953cccb7 100644 --- a/src/libstd/cmp.rs +++ b/src/libstd/cmp.rs @@ -14,16 +14,16 @@ use core::f64; use core::float; -const fuzzy_epsilon: float = 1.0e-6; +pub const FUZZY_EPSILON: float = 1.0e-6; -pub trait FuzzyEq { +pub trait FuzzyEq { pure fn fuzzy_eq(&self, other: &Self) -> bool; - pure fn fuzzy_eq_eps(&self, other: &Self, epsilon: &Self) -> bool; + pure fn fuzzy_eq_eps(&self, other: &Self, epsilon: &Eps) -> bool; } -impl float: FuzzyEq { +impl float: FuzzyEq { pure fn fuzzy_eq(&self, other: &float) -> bool { - self.fuzzy_eq_eps(other, &fuzzy_epsilon) + self.fuzzy_eq_eps(other, &FUZZY_EPSILON) } pure fn fuzzy_eq_eps(&self, other: &float, epsilon: &float) -> bool { @@ -31,9 +31,9 @@ impl float: FuzzyEq { } } -impl f32: FuzzyEq { +impl f32: FuzzyEq { pure fn fuzzy_eq(&self, other: &f32) -> bool { - self.fuzzy_eq_eps(other, &(fuzzy_epsilon as f32)) + self.fuzzy_eq_eps(other, &(FUZZY_EPSILON as f32)) } pure fn fuzzy_eq_eps(&self, other: &f32, epsilon: &f32) -> bool { @@ -41,9 +41,9 @@ impl f32: FuzzyEq { } } -impl f64: FuzzyEq { +impl f64: FuzzyEq { pure fn fuzzy_eq(&self, other: &f64) -> bool { - self.fuzzy_eq_eps(other, &(fuzzy_epsilon as f64)) + self.fuzzy_eq_eps(other, &(FUZZY_EPSILON as f64)) } pure fn fuzzy_eq_eps(&self, other: &f64, epsilon: &f64) -> bool { @@ -63,3 +63,40 @@ fn test_fuzzy_eq_eps() { assert (&1.2f).fuzzy_eq_eps(&0.9, &0.5); assert !(&1.5f).fuzzy_eq_eps(&0.9, &0.5); } + +#[test] +mod test_complex{ + use cmp::*; + + struct Complex { r: float, i: float } + + impl Complex: FuzzyEq { + pure fn fuzzy_eq(&self, other: &Complex) -> bool { + self.fuzzy_eq_eps(other, &FUZZY_EPSILON) + } + + pure fn fuzzy_eq_eps(&self, other: &Complex, + epsilon: &float) -> bool { + self.r.fuzzy_eq_eps(&other.r, epsilon) && + self.i.fuzzy_eq_eps(&other.i, epsilon) + } + } + + #[test] + fn test_fuzzy_equals() { + let a = Complex {r: 0.9, i: 0.9}; + let b = Complex {r: 0.9, i: 0.9}; + + assert (a.fuzzy_eq(&b)); + } + + #[test] + fn test_fuzzy_eq_eps() { + let other = Complex {r: 0.9, i: 0.9}; + + assert (&Complex {r: 0.9, i: 1.2}).fuzzy_eq_eps(&other, &0.5); + assert (&Complex {r: 1.2, i: 0.9}).fuzzy_eq_eps(&other, &0.5); + assert !(&Complex {r: 0.9, i: 1.5}).fuzzy_eq_eps(&other, &0.5); + assert !(&Complex {r: 1.5, i: 0.9}).fuzzy_eq_eps(&other, &0.5); + } +} diff --git a/src/test/run-pass/trait-inheritance-num.rs b/src/test/run-pass/trait-inheritance-num.rs index 90e7db7dbbb..ca720c27d77 100644 --- a/src/test/run-pass/trait-inheritance-num.rs +++ b/src/test/run-pass/trait-inheritance-num.rs @@ -18,7 +18,7 @@ pub trait NumExt: Num Eq Ord {} -pub trait FloatExt: NumExt FuzzyEq {} +pub trait FloatExt: NumExt FuzzyEq {} fn greater_than_one(n: &T) -> bool { *n > from_int(1) } fn greater_than_one_float(n: &T) -> bool { *n > from_int(1) } diff --git a/src/test/run-pass/trait-inheritance-num2.rs b/src/test/run-pass/trait-inheritance-num2.rs index 455b3946ec0..ecedaac8daa 100644 --- a/src/test/run-pass/trait-inheritance-num2.rs +++ b/src/test/run-pass/trait-inheritance-num2.rs @@ -94,7 +94,7 @@ pub impl i64: IntegerExt {} pub impl int: IntegerExt {} -pub trait FloatExt: NumExt FuzzyEq {} +pub trait FloatExt: NumExt FuzzyEq {} pub impl f32: FloatExt {} pub impl f64: FloatExt {}