| extern crate overload; | |
| use overload::overload; | |
| use std::ops; | |
| #[derive(PartialEq, Debug)] | |
| struct A(i32); | |
| #[derive(PartialEq, Debug)] | |
| struct B(i32); | |
| #[derive(PartialEq, Debug)] | |
| struct C(i32); | |
| overload!((a: A) + (b: B) -> C { C(a.0 + b.0) }); | |
| #[test] | |
| fn add() { | |
| assert_eq!(A(3) + B(4), C(3 + 4)); | |
| } | |
| overload!((a: A) - (b: B) -> C { C(a.0 - b.0) }); | |
| #[test] | |
| fn sub() { | |
| assert_eq!(A(3) - B(4), C(3 - 4)); | |
| } | |
| overload!((a: A) * (b: B) -> C { C(a.0 * b.0) }); | |
| #[test] | |
| fn mul() { | |
| assert_eq!(A(3) * B(4), C(3 * 4)); | |
| } | |
| overload!((a: A) / (b: B) -> C { C(a.0 / b.0) }); | |
| #[test] | |
| fn div() { | |
| assert_eq!(A(6) / B(3), C(6 / 3)); | |
| } | |
| overload!((a: A) % (b: B) -> C { C(a.0 % b.0) }); | |
| #[test] | |
| fn rem() { | |
| assert_eq!(A(6) % B(4), C(6 % 4)); | |
| } | |
| overload!((a: A) & (b: B) -> C { C(a.0 & b.0) }); | |
| #[test] | |
| fn bitand() { | |
| assert_eq!(A(6) & B(4), C(6 & 4)); | |
| } | |
| overload!((a: A) | (b: B) -> C { C(a.0 | b.0) }); | |
| #[test] | |
| fn bitor() { | |
| assert_eq!(A(6) | B(4), C(6 | 4)); | |
| } | |
| overload!((a: A) ^ (b: B) -> C { C(a.0 ^ b.0) }); | |
| #[test] | |
| fn bitxor() { | |
| assert_eq!(A(6) ^ B(4), C(6 ^ 4)); | |
| } | |
| overload!((a: A) << (b: B) -> C { C(a.0 << b.0) }); | |
| #[test] | |
| fn shl() { | |
| assert_eq!(A(6) << B(4), C(6 << 4)); | |
| } | |
| overload!((a: A) >> (b: B) -> C { C(a.0 >> b.0) }); | |
| #[test] | |
| fn shr() { | |
| assert_eq!(A(6) >> B(4), C(6 >> 4)); | |
| } |