PyTest - write your own fixture