Creating a data-driven test in Python using DDT