Updated tests to use pytest fixtures