Actually mlflow is primarily for experiment tracking and model management. For testing you can use pytest & tox.

Check my latest blog on pytest & tox - https://pub.towardsai.net/boost-your-code-quality-with-pytest-tox-3d0e2c9c334e