I took an app Android-2048 as a sample to show how AndroidScreenshotTestCase works.
$ pip install mock
$ pip install Pillow
$ ./gradlew recordDebugAndroidTestScreenshotTest
$ ./gradlew executeScreenshotTests -Precord
$ ./gradlew verifyDebugAndroidTestScreenshotTest
$ ./gradlew executeScreenshotTests