Add some unit tests using py.test

- we need to have more