Refine test suite

Use `pytest` features for more clean and lean test suite.