Fix shows_time_by_default TimeText test.

Bug: 376698283
Test: N/A
Relnote: N/A
Change-Id: Ic0c115373d878e00c75346f49b207581740c6cb1
diff --git a/wear/compose/compose-material3/src/androidTest/kotlin/androidx/wear/compose/material3/TimeTextTest.kt b/wear/compose/compose-material3/src/androidTest/kotlin/androidx/wear/compose/material3/TimeTextTest.kt
index d7cd4f0..f932e4a 100644
--- a/wear/compose/compose-material3/src/androidTest/kotlin/androidx/wear/compose/material3/TimeTextTest.kt
+++ b/wear/compose/compose-material3/src/androidTest/kotlin/androidx/wear/compose/material3/TimeTextTest.kt
@@ -53,11 +53,11 @@
     }
 
     @Test
-    fun shows_time_by_default() {
-        val timeText = "test"
+    fun shows_time_by_default_on_non_round_device() {
+        val timeText = "time"
 
-        rule.setContent {
-            MaterialTheme() {
+        rule.setContentWithTheme {
+            DeviceConfigurationOverride(DeviceConfigurationOverride.RoundScreen(false)) {
                 TimeText(
                     timeSource =
                         object : TimeSource {
@@ -66,7 +66,9 @@
                 )
             }
         }
-        rule.onNodeWithText(timeText).assertExists()
+
+        // Note that onNodeWithText doesn't work for curved text, so only testing for non-round.
+        rule.onNodeWithText(timeText).assertIsDisplayed()
     }
 
     @Test