Use pytest-style setup