Add coverage support for pytest