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));
}