summaryrefslogtreecommitdiffstats
path: root/vendor/overload/tests/assignment.rs
blob: 9657b4bd4dcfc79a10998e3ad10aa419c9212046 (plain)
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);
}