1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
|
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));
}
|