Add tests

Why don't you use `self.assertNotEqual` instead of `assert` ?