Last I checked (October) I could not get the build.gradle to play nicely. Most of the other material online involved using instrumentCompile to trick Android Studio into running your unit tests and a custom runner/plugin: https://github.com/JakeWharton/gradle-android-test-plugin
If you have gotten it working in Android Studio, I'd be very keen to see how (if you are willing to share).