From a154c98faa3290e9ca45a0e8dabd39c0d74842fc Mon Sep 17 00:00:00 2001
From: tofunmi <tofunmi@google.com>
Date: Thu, 11 Apr 2024 03:42:24 -0700
Subject: [PATCH] Fix test setup for exportUltraHdrImageThenHdrVideo_exportsHdr

make sure compostion.experimentalSetRetainHdrFromUltraHdrImage(true) so that we get hdr output

PiperOrigin-RevId: 623773784
---
 .../media3/transformer/mh/TransformerMhUltraHdrTest.java      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/TransformerMhUltraHdrTest.java b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/TransformerMhUltraHdrTest.java
index a8301710c3..ece2c8966d 100644
--- a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/TransformerMhUltraHdrTest.java
+++ b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/TransformerMhUltraHdrTest.java
@@ -115,8 +115,8 @@ public final class TransformerMhUltraHdrTest {
   public void exportUltraHdrImageThenHdrVideo_exportsHdr() throws Exception {
     assumeDeviceSupportsUltraHdrEditing();
     Composition composition =
-        createComposition(
-            /* presentation= */ null,
+        createUltraHdrComposition(
+            /* tonemap= */ false,
             oneFrameFromImage(ULTRA_HDR_URI_STRING, NO_EFFECT),
             clippedVideo(MP4_ASSET_1080P_5_SECOND_HLG10, NO_EFFECT, ONE_FRAME_END_POSITION_MS));