VScode Generate Unit Tests feature

Hi,

When using the “Generate Unit Tests” feature, I would want to force the model to generate tests with a particular framework. For instance, pytest for Python code. As for now, it generates tests with the unittest framework (I tried with a few models, they all tend to use unittest).

I guess enabling the user to add a custom prompt would be a way to allow that.