Made tests cleaner with pytest.parametrize