I use the Neovim text editor with PyRight and MyPy Language Server plugins, but that’s a whole rabbit hole of basically relearning how to edit code. You probably shouldn’t mess with that JUST for testing. There’s a notoriously steep learning curve. I still think everybody should give it a try through! Once you learn it, it’s like magic!
Ok, stepping off the soapbox
PyRight is the LS that comes with vscode, so you could definitely test there.
And MyPy is the reference implementation of python static type checking. I was able to easily find a vscode plugin for that.
You may also want to test with Pyre from Facebook, but that seems a bit more involved to set up.