Add tests

it's ok to use plain asserts with py.test, though I don't really mind to use assertNotEqual() here