I use pytest/selenium to set up a test infrastructure for my UI.
It was built off of this example:
Although , it doesn’t work out of the box, you’ll have to do some tweaking. But it was a great start and saved me loads of time.