Applies Material You Colors in fragment rule.

Test: http://ab/I77900010255425277
Change-Id: Ib09e8ab73e4f9baa10fa11a7a70df4bcf6f9c13c
diff --git a/libraries/screenshot/src/main/java/platform/test/screenshot/FragmentScreenshotTestRule.kt b/libraries/screenshot/src/main/java/platform/test/screenshot/FragmentScreenshotTestRule.kt
index ab7c644..da8a916 100644
--- a/libraries/screenshot/src/main/java/platform/test/screenshot/FragmentScreenshotTestRule.kt
+++ b/libraries/screenshot/src/main/java/platform/test/screenshot/FragmentScreenshotTestRule.kt
@@ -47,7 +47,7 @@
     private val commonRule =
         RuleChain.outerRule(deviceEmulationRule).around(screenshotRule).around(activityRule)
     private val deviceRule = RuleChain.outerRule(colorsRule).around(commonRule)
-    private val roboRule = RuleChain.outerRule(timeZoneRule).around(commonRule)
+    private val roboRule = RuleChain.outerRule(colorsRule).around(timeZoneRule).around(commonRule)
     private val isRobolectric = if (Build.FINGERPRINT.contains("robolectric")) true else false
 
     override fun apply(base: Statement, description: Description): Statement {
@@ -99,19 +99,8 @@
             beforeScreenshot(activity)
         }
 
-        val originalBitmap = contentView?.captureToBitmapAsync()?.get(10, TimeUnit.SECONDS)
-        if (originalBitmap == null) {
-            error("timeout while trying to capture view to bitmap")
-        }
-        return if (isRobolectric) {
-            bitmapWithMaterialYouColorsSimulation(
-                originalBitmap,
-                emulationSpec.isDarkTheme,
-                /* doPixelAveraging= */ true
-            )
-        } else {
-            originalBitmap
-        }
+        return contentView?.captureToBitmapAsync()?.get(10, TimeUnit.SECONDS)
+            ?: error("timeout while trying to capture view to bitmap")
     }
 
     /**