replace deprecated unittest method
Merge pull request #33 from tshepang/diff

replace deprecated unittest method