One way of doing this is to do add junit.jar to your $CLASSPATH
as an external dependency.

So to do that, go to project structure, and then add JUnit as one of the libraries as shown in the gif.
In the 'Choose Modules' prompt choose only the modules that you'd need JUnit for.