If it were me, I would put new tests along with the old ones.

p.s. please run flake8 and remove unused imports