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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
|
extern crate overload;
use overload::overload;
use std::ops;
#[derive(PartialEq, Debug)]
struct A(i32);
#[derive(PartialEq, Debug)]
struct B(i32);
overload!((a: &mut A) += (b: B) { a.0 += b.0; });
#[test]
fn add_assign() {
let mut a = A(3);
a += B(4);
assert_eq!(A(3 + 4), a);
}
overload!((a: &mut A) -= (b: B) { a.0 -= b.0; });
#[test]
fn sub_assign() {
let mut a = A(3);
a -= B(4);
assert_eq!(A(3 - 4), a);
}
overload!((a: &mut A) *= (b: B) { a.0 *= b.0; });
#[test]
fn mul_assign() {
let mut a = A(3);
a *= B(4);
assert_eq!(A(3 * 4), a);
}
overload!((a: &mut A) /= (b: B) { a.0 /= b.0; });
#[test]
fn div_assign() {
let mut a = A(6);
a /= B(3);
assert_eq!(A(6 / 3), a);
}
overload!((a: &mut A) %= (b: B) { a.0 %= b.0; });
#[test]
fn rem_assign() {
let mut a = A(6);
a %= B(4);
assert_eq!(A(6 % 4), a);
}
overload!((a: &mut A) &= (b: B) { a.0 &= b.0; });
#[test]
fn bitand_assign() {
let mut a = A(6);
a &= B(4);
assert_eq!(A(6 & 4), a);
}
overload!((a: &mut A) |= (b: B) { a.0 |= b.0; });
#[test]
fn bitor_assign() {
let mut a = A(6);
a |= B(4);
assert_eq!(A(6 | 4), a);
}
overload!((a: &mut A) ^= (b: B) { a.0 ^= b.0; });
#[test]
fn bitxor_assign() {
let mut a = A(6);
a ^= B(4);
assert_eq!(A(6 ^ 4), a);
}
overload!((a: &mut A) <<= (b: B) { a.0 <<= b.0; });
#[test]
fn shl_assign() {
let mut a = A(6);
a <<= B(4);
assert_eq!(A(6 << 4), a);
}
overload!((a: &mut A) >>= (b: B) { a.0 >>= b.0; });
#[test]
fn shr_assign() {
let mut a = A(6);
a >>= B(4);
assert_eq!(A(6 >> 4), a);
}
|