Parameterized tests might reduce the amount of test code. In today's task, the code is already working. Your responsibility is to rewrite the tests, so they will use data sets. Preferably, all test cases should be still covered, but there should be only two tests.