Pruebas de sabores e instrumentación - configuración de gradle

Utilizo sabores en mi proyecto y quiero agregar pruebas de instrumentación adicionales para cada sabor.

Entonces creé

MyApplication/src/androidTestFlavor1/java/com.package.test
MyApplication/src/androidTestFlavor2/java/com.package.test

Pero no funciona correctamente.

Entonces, intenté configurarlo en build.gradle, agregué

android {
    ...

    sourceSets {
        flavor1{
            instrumentTest.setRoot('src/instrumentationTestFlavor/java')
        }
    }

    ...
}

pero me sale un error:

Error: (59, 0) No se pudo encontrar la propiedad 'instrumentTest' en la fuente establecida ding.

¿Cuál es la mejor solución aquí?