| import friends_operator_overloading |
| |
| friends_operator_overloading.sanity_checker_ShiftA() |
| friends_operator_overloading.sanity_checker_ShiftB() |
| |
| sa1 = friends_operator_overloading.ShiftA(200) |
| sa2 = friends_operator_overloading.ShiftA(1000) |
| sb1 = friends_operator_overloading.ShiftB(200) |
| sb2 = friends_operator_overloading.ShiftB(1000) |
| |
| # Shift operator via members |
| sa3 = sa2 << sa1 |
| val = sa3.getVal() |
| if val != 800: |
| raise RuntimeError("Wrong val: {}".format(val)) |
| |
| sa4 = sa2 << 300 |
| val = sa4.getVal() |
| if val != 700: |
| raise RuntimeError("Wrong val: {}".format(val)) |
| |
| sb3 = sb2 << sb1 |
| val = sb3.getVal() |
| if val != 800: |
| raise RuntimeError("Wrong val: {}".format(val)) |
| |
| sb4 = sb2 << 300 |
| val = sb4.getVal() |
| if val != 700: |
| raise RuntimeError("Wrong val: {}".format(val)) |
| |
| # Shift operator via global wrapper |
| shift = friends_operator_overloading.__lshift__(sa2, sa1) |
| val = shift.getVal() |
| if val != 800: |
| raise RuntimeError("Wrong val: {}".format(val)) |
| |
| shift = friends_operator_overloading.__lshift__(sb2, sb1) |
| val = shift.getVal() |
| if val != 800: |
| raise RuntimeError("Wrong val: {}".format(val)) |