#2 rewrite tests to py.test

I had rewrite tests :taxi: