Fixes #4

- Add 2 assertions in the test

- Directly calculate new leakage, rather than get the leakage count from
the MDNetwork and then subtract. This should save some time.