Use pytest parametrization