chore: little bit of cleanup to a test module, use parametrization

chore: little bit of cleanup to a test module, use parametrization #255